diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index d78f63a..eddb933 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -17,13 +17,13 @@ jobs:
 
     steps:
       - uses: actions/checkout@v3
-      - name: Setup upterm session
-        uses: lhotari/action-upterm@v1
-        with:
-          ## limits ssh access and adds the ssh public key for the user which triggered the workflow
-          limit-access-to-actor: true
-          ## limits ssh access and adds the ssh public keys of the listed GitHub users
-          limit-access-to-users: rmontanana
+      # - name: Setup upterm session
+      #   uses: lhotari/action-upterm@v1
+      #   with:
+      #     ## limits ssh access and adds the ssh public key for the user which triggered the workflow
+      #     limit-access-to-actor: true
+      #     ## limits ssh access and adds the ssh public keys of the listed GitHub users
+      #     limit-access-to-users: rmontanana
       - name: Set up Python ${{ matrix.python }}
         uses: actions/setup-python@v2
         with:
diff --git a/benchmark/Models.py b/benchmark/Models.py
index cee25ba..d2e8db0 100644
--- a/benchmark/Models.py
+++ b/benchmark/Models.py
@@ -8,7 +8,7 @@ from sklearn.ensemble import (
 )
 from sklearn.svm import SVC
 from stree import Stree
-from bayesclass import TAN, KDB, AODE
+from bayesclass.bayesclass import TAN, KDB, AODE
 from wodt import Wodt
 from odte import Odte
 from xgboost import XGBClassifier
diff --git a/requirements.txt b/requirements.txt
index fd384e7..cba834e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ odte
 cython
 mdlp-discretization
 mufs
-#bayesclass @ git+ssh://git@github.com/doctorado-ml/bayesclass.git
+bayesclass @ git+ssh://git@github.com/doctorado-ml/bayesclass.git
 xlsxwriter
 openpyxl
 tqdm