diff --git a/.github/actions/cognee_setup/action.yml b/.github/actions/cognee_setup/action.yml
index 9ae669b53..77c5b3b1f 100644
--- a/.github/actions/cognee_setup/action.yml
+++ b/.github/actions/cognee_setup/action.yml
@@ -24,7 +24,7 @@ runs:
       uses: astral-sh/setup-uv@v4
       with:
         enable-cache: true
-    
+
     - name: Rebuild uv lockfile
       shell: bash
       run: |
@@ -41,4 +41,4 @@ runs:
             EXTRA_ARGS="$EXTRA_ARGS --extra $extra"
           done
         fi
-        uv sync --extra api --extra docs --extra evals --extra gemini --extra codegraph --extra ollama --extra dev --extra neo4j $EXTRA_ARGS
+        uv sync --extra api --extra docs --extra evals --extra gemini --extra codegraph --extra ollama --extra dev --extra baml --extra neo4j $EXTRA_ARGS
diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml
index bfe15a6cf..e2264da3d 100644
--- a/.github/workflows/basic_tests.yml
+++ b/.github/workflows/basic_tests.yml
@@ -189,9 +189,6 @@ jobs:
         with:
           python-version: ${{ inputs.python-version }}
 
-      - name: Install BAML Dependencies
-        run: uv add "baml-py>=0.201.0,<0.202.0"
-
       - name: Run Simple Examples
         run: uv run python ./examples/python/simple_example.py