diff --git a/.github/unittest/install_dependencies.sh b/.github/unittest/install_dependencies.sh
index f52ce9d..0bf44c6 100644
--- a/.github/unittest/install_dependencies.sh
+++ b/.github/unittest/install_dependencies.sh
@@ -4,7 +4,7 @@ python -m pip install --upgrade pip
 # Install dependencies
 python -m pip install flake8 pytest pytest-cov hydra-core tqdm
 python -m pip install torch torchvision
-python -m pip install transformers promptsmiles torchrl rdkit==2023.3.3 MolScore # causal-conv1d>=1.4.0 mamba-ssm==1.2.2
+python -m pip install transformers promptsmiles torchrl rdkit==2023.3.3 MolScore
 python -m pip install deepsmiles
 python -m pip install selfies
 python -m pip install smi2sdf
@@ -13,6 +13,8 @@ python -m pip install atomInSmiles
 python -m pip install safe-mol
 python -m pip install smizip
 python -m pip install molbloom
+python -m pip causal-conv1d>=1.4.0 
+python -m pip mamba-ssm==1.2.2
 
 # Verify installations
 python -c "import transformers; print(transformers.__version__)"
diff --git a/.github/unittest/install_dependencies_nightly.sh b/.github/unittest/install_dependencies_nightly.sh
index 0d589ea..0366e49 100644
--- a/.github/unittest/install_dependencies_nightly.sh
+++ b/.github/unittest/install_dependencies_nightly.sh
@@ -13,6 +13,8 @@ python -m pip install atomInSmiles
 python -m pip install safe-mol
 python -m pip install smizip
 python -m pip install molbloom
+python -m pip causal-conv1d>=1.4.0 
+python -m pip mamba-ssm==1.2.2
 
 # Verify installations
 python -c "import transformers; print(transformers.__version__)"