diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 30ff3c3..3842902 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -35,7 +35,7 @@ jobs:
             find . -type f > before_test_files.txt
             conda env create -f environment.yml
             conda activate atomgpt
-            conda install pytest coverage codecov 
+            conda install pytest coverage codecov  alignn -y
             #conda install pytest coverage codecov pandas numpy matplotlib phonopy scikit-learn jarvis-tools --quiet
             #export DGLBACKEND=pytorch
             #export CUDA_VISIBLE_DEVICES="-1"
diff --git a/setup.py b/setup.py
index aa12453..0a407e4 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
         "pydantic_settings",
         "peft",
         "trl",
-        "alignn",
+        #"alignn",
         "triton",
         "torch",