diff --git a/h1st/model/ml/xgboost/classifier.py b/h1st/model/ml/xgboost/classifier.py
index 16b50283..384d7726 100644
--- a/h1st/model/ml/xgboost/classifier.py
+++ b/h1st/model/ml/xgboost/classifier.py
@@ -2,7 +2,7 @@
 
 
 class XGBClassifierModel(XGBRegressionModel):
-    name = 'XGBClassifierModel'
+    name = 'XGBClassifier'
 
     def predict(self, input_data: dict) -> dict:
         pred = super().predict(input_data)[self.output_key]
diff --git a/h1st/model/ml/xgboost/regression.py b/h1st/model/ml/xgboost/regression.py
index 752e14b8..92de7045 100644
--- a/h1st/model/ml/xgboost/regression.py
+++ b/h1st/model/ml/xgboost/regression.py
@@ -16,7 +16,7 @@ class XGBRegressionModel(Model):
 
     input_key = 'X'
     output_key = 'predictions'
-    name = 'XGBRegressionModel'
+    name = 'XGBRegression'
 
     def __init__(
         self,
diff --git a/pyproject.toml b/pyproject.toml
index 5f6fb932..49c423c5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "h1st"
-version = "2.1.0"
+version = "2.1.1"
 description = "Human-First AI (H1st)"
 authors = ["Aitomatic, Inc. <engineering@aitomatic.com>"]
 license = "Apache-2.0"
diff --git a/setup.cfg b/setup.cfg
index 889c765a..1efbcae9 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -4,4 +4,4 @@
 
 [metadata]
 name = h1st
-version = 2.1.0
+version = 2.1.1