diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index 345ca77..5f212a4 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -30,8 +30,8 @@ jobs:
       - name: Run examples 
         run: pip install .&&find examples -maxdepth 2 -name "*.py" -exec  python3 {} \;
 
-      - name: Publish to PyPI                
-        uses: pypa/gh-action-pypi-publish@release/v1
-        with:
-          password: ${{ secrets.PYPI_GLOBAL_UB }}
-          repository-url: https://upload.pypi.org/legacy/
+      #- name: Publish to PyPI                
+      #  uses: pypa/gh-action-pypi-publish@release/v1
+      #  with:
+      #    password: ${{ secrets.PYPI_GLOBAL_UB }}
+      #    repository-url: https://upload.pypi.org/legacy/
diff --git a/examples/regression.py b/examples/regression.py
index 15f9f54..8d041a9 100644
--- a/examples/regression.py
+++ b/examples/regression.py
@@ -11,10 +11,10 @@
 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
 
 # Initialize the unified regr (example with XGBoost)
-regr1 = ub.GBDTregr(model_type='xgboost')
-#regr2 = ub.GBDTregr(model_type='catboost')
-regr3 = ub.GBDTregr(model_type='lightgbm')
-regr4 = ub.GBDTregr(model_type='gradientboosting', 
+regr1 = ub.GBDTRegressor(model_type='xgboost')
+#regr2 = ub.GBDTRegressor(model_type='catboost')
+regr3 = ub.GBDTRegressor(model_type='lightgbm')
+regr4 = ub.GBDTRegressor(model_type='gradientboosting', 
                     colsample=0.9)
 
 # Fit the model