diff --git a/tests/python/test_basic.py b/tests/python/test_basic.py
index 081145e2c..926fe57b2 100644
--- a/tests/python/test_basic.py
+++ b/tests/python/test_basic.py
@@ -184,10 +184,10 @@ class TestBasic(unittest.TestCase):
 
     def test_load_file_invalid(self):
 
-        self.assertRaises(ValueError, xgb.Booster,
+        self.assertRaises(xgb.XGBoostError, xgb.Booster,
                           model_file='incorrect_path')
 
-        self.assertRaises(ValueError, xgb.Booster,
+        self.assertRaises(xgb.XGBoostError, xgb.Booster,
                           model_file=u'不正なパス')
 
     def test_dmatrix_numpy_init(self):