diff --git a/demo/gpu_acceleration/bosch.py b/demo/gpu_acceleration/bosch.py
index f717532cc..2294b24e7 100644
--- a/demo/gpu_acceleration/bosch.py
+++ b/demo/gpu_acceleration/bosch.py
@@ -3,7 +3,7 @@ import pandas as pd
 import xgboost as xgb
 import time
 import random
-from sklearn.cross_validation import StratifiedKFold
+from sklearn.model_selection import StratifiedKFold
 
 #For sampling rows from input file
 random_seed = 9