diff --git a/python/xgboost.py b/python/xgboost.py
index d7cf9f63e21d..3be5a2b066e5 100644
--- a/python/xgboost.py
+++ b/python/xgboost.py
@@ -22,6 +22,7 @@ class REntry(ctypes.Structure):
 xglib.XGDMatrixGetLabel.restype =  ctypes.POINTER( ctypes.c_float )
 xglib.XGDMatrixGetWeight.restype =  ctypes.POINTER( ctypes.c_float )
 xglib.XGDMatrixGetRow.restype = ctypes.POINTER( REntry )
+xglib.XGBoosterCreate.restype = ctypes.c_void_p
 xglib.XGBoosterPredict.restype = ctypes.POINTER( ctypes.c_float ) 
 
 def ctypes2numpy( cptr, length ):