diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml
index b6da0d295..bfe15a6cf 100644
--- a/.github/workflows/basic_tests.yml
+++ b/.github/workflows/basic_tests.yml
@@ -190,7 +190,7 @@ jobs:
           python-version: ${{ inputs.python-version }}
 
       - name: Install BAML Dependencies
-        run: uv add "cognee[baml]"
+        run: uv add "baml-py>=0.201.0,<0.202.0"
 
       - name: Run Simple Examples
         run: uv run python ./examples/python/simple_example.py