diff --git a/.github/workflows/django-test.yml b/.github/workflows/django-test.yml
index 6f2b937..b9244a9 100644
--- a/.github/workflows/django-test.yml
+++ b/.github/workflows/django-test.yml
@@ -1,14 +1,18 @@
-name: Django Test
+name: Test Datalab
 
 on:
   push:
+    branches: [ "main" ]
   pull_request:
     branches: [ "main" ]
 
 jobs:
-  build:
+  test:
 
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
     
     steps:
     - name: Checkout
@@ -16,7 +20,7 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v3
       with: 
-        python-version: '3.11'
+        python-version: ${{ matrix.python-version }}
     - name: Install Poetry
       run: |
         python -m pip install --upgrade pip
diff --git a/datalab/datalab_session/data_operations/utils.py b/datalab/datalab_session/data_operations/utils.py
index a3bf4c1..13d103d 100644
--- a/datalab/datalab_session/data_operations/utils.py
+++ b/datalab/datalab_session/data_operations/utils.py
@@ -1,5 +1,6 @@
 from pkgutil import walk_packages
 import inspect
+from importlib import import_module
 from django.utils.module_loading import import_string
 from datalab.datalab_session import data_operations
 
@@ -8,7 +9,7 @@ def available_operations():
     operations = {}
     base_operation = import_string('datalab.datalab_session.data_operations.data_operation.BaseDataOperation')
     for (loader, module_name, _) in walk_packages(data_operations.__path__):
-        module = loader.find_module(module_name).load_module()
+        module = import_module(f'{data_operations.__name__}.{module_name}')
         members = inspect.getmembers(module, inspect.isclass)
         for member in members:
             if member[0] != 'BaseDataOperation' and issubclass(member[1], base_operation):