diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 83e9f2da..e7122845 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -16,18 +16,16 @@ jobs:
   test:
     name: "Run unit tests for reference code"
     runs-on: ubuntu-latest
+    container:
+      image: sagemath/sagemath:latest
+      options: --user root
     steps:
     - name: Checkout
       uses: actions/checkout@v3
-      with:
-        submodules: true
 
-    - name: Install Sage
-      run: |
-        sudo apt-get update
-        sudo apt-get install -y sagemath python3-cffi
-        sage -pip install pycryptodomex
+    - name: Install dependencies
+      run: sage --pip install pycryptodomex
 
     - name: Run tests
       working-directory: poc
-      run: make test
+      run: sage -python -m unittest