diff --git a/python-package/xgboost/compat.py b/python-package/xgboost/compat.py
index cc035b6da..e94c67b84 100644
--- a/python-package/xgboost/compat.py
+++ b/python-package/xgboost/compat.py
@@ -48,3 +48,5 @@ except ImportError:
     XGBModelBase = object
     XGBClassifierBase = object
     XGBRegressorBase = object
+    XGBKFold = None
+    XGBStratifiedKFold = None