diff --git a/.github/workflows/test_basics.yml b/.github/workflows/test_basics.yml
index a52b586c..3958c0aa 100644
--- a/.github/workflows/test_basics.yml
+++ b/.github/workflows/test_basics.yml
@@ -2,11 +2,25 @@ name: test-basics
 on: [push, pull_request, workflow_dispatch]
 
 jobs:
-  Ubuntu:
-    strategy:
-      matrix:
-        modules: [01_write_ivy_code, 02_unify_code, 03_compile_code, 04_transpile_code, 05_lazy_vs_eager, 06_how_to_use_decorators, 07_transpile_any_library, 08_transpile_any_model, 09_write_a_model_using_ivy]
+  SetupDocker:
+    runs-on: ubuntu-20.04
+    steps:
+      - name: Checkout Demos🛎
+        uses: actions/checkout@v2
+        with:
+          path: demos
+          persist-credentials: false
+          submodules: "recursive"
+          fetch-depth: 100
+
+      - name: Set Up Docker Container
+        id: setup-docker
+        run: |
+          cd demos
+          docker run -d --name ivy-container unifyai/ivy:latest tail -f /dev/null
 
+  RunTests:
+    needs: SetupDocker
     runs-on: ubuntu-20.04
     steps:
       - name: Checkout Demos🛎
@@ -21,8 +35,16 @@ jobs:
         id: tests
         run: |
           cd demos
-          docker run -v `pwd`:/ivy/demos unifyai/ivy:latest demos/tests/test_demos.sh ${{ secrets.USER_API_KEY }} ${{ matrix.modules }}.ipynb "basics"
+          docker exec ivy-container /ivy/demos/tests/test_demos.sh ${{ secrets.USER_API_KEY }} ${{ matrix.test_files }} "basics"
 
       - name: Check on Failures
         if: steps.tests.outcome != 'success'
         run: exit 1
+
+    strategy:
+      matrix:
+        include:
+          - test_files: "01_write_ivy_code.ipynb"
+          - test_files: "02_unify_code.ipynb"
+          - test_files: "03_compile_code.ipynb"
+          # Add more test files here