diff --git a/.github/workflows/profiling.yaml b/.github/workflows/profiling.yaml
index e1ac05a9d..175d824a7 100644
--- a/.github/workflows/profiling.yaml
+++ b/.github/workflows/profiling.yaml
@@ -29,7 +29,7 @@ jobs:
     - name: Install dependencies
       run: |
         poetry install --no-interaction --all-extras
-        poetry add scalene 
+        poetry add pyinstrument 
 
 
     # Set environment variables for SHAs
@@ -50,7 +50,7 @@ jobs:
         # Ensure the script is executable
         chmod +x cognee/api/v1/cognify/code_graph_pipeline.py
         # Run Scalene
-        poetry run scalene --json --outfile base_results.json cognee/api/v1/cognify/code_graph_pipeline.py
+        poetry run pyinstrument --json --outfile base_results.json cognee/api/v1/cognify/code_graph_pipeline.py
 
     # Run profiler on head branch
     - name: Run profiler on head branch
@@ -64,7 +64,7 @@ jobs:
         # Ensure the script is executable
         chmod +x cognee/api/v1/cognify/code_graph_pipeline.py
         # Run Scalene
-        poetry run scalene --json --outfile head_results.json cognee/api/v1/cognify/code_graph_pipeline.py
+        poetry run pyinstrument --json --outfile head_results.json cognee/api/v1/cognify/code_graph_pipeline.py
 
     # Compare profiling results
     - name: Compare profiling results