diff --git a/src/PyClassifiers/XGBoost.cc b/src/PyClassifiers/XGBoost.cc
index a073b1a..e1bcfb6 100644
--- a/src/PyClassifiers/XGBoost.cc
+++ b/src/PyClassifiers/XGBoost.cc
@@ -2,12 +2,8 @@
 
 //See https ://stackoverflow.com/questions/36071672/using-xgboost-in-c
 namespace pywrap {
-    XGBoost::XGBoost() : PyClassifier("xgboost", "XGBClassifier")
+    XGBoost::XGBoost() : PyClassifier("xgboost", "XGBClassifier", true)
     {
         validHyperparameters = { "tree_method", "early_stopping_rounds", "n_jobs" };
     }
-    std::string XGBoost::version()
-    {
-        return callMethodString("1.0");
-    }
 } /* namespace pywrap */
\ No newline at end of file
diff --git a/src/PyClassifiers/XGBoost.h b/src/PyClassifiers/XGBoost.h
index b384ded..930b92e 100644
--- a/src/PyClassifiers/XGBoost.h
+++ b/src/PyClassifiers/XGBoost.h
@@ -7,7 +7,6 @@ namespace pywrap {
     public:
         XGBoost();
         ~XGBoost() = default;
-        std::string version();
     };
 } /* namespace pywrap */
 #endif /* XGBOOST_H */
\ No newline at end of file