diff --git a/.github/workflows/check_unit_tests.yml b/.github/workflows/check_unit_tests.yml
index e8fd8a76..98b61ae0 100644
--- a/.github/workflows/check_unit_tests.yml
+++ b/.github/workflows/check_unit_tests.yml
@@ -17,7 +17,6 @@ jobs:
           - 5432:5432
         env:
           POSTGRES_PASSWORD: pw
-          POSTGRES_DB: testdb
 
     steps:
       - name: Checkout code
@@ -40,11 +39,6 @@ jobs:
 
       - name: Run unit tests
         env:
-          MPI_DB_TYPE: postgres
-          MPI_DBNAME: testdb
-          MPI_HOST: localhost
-          MPI_PORT: 5432
-          MPI_USER: postgres
-          MPI_PASSWORD: pw
+          DB_URI: "postgresql+psycopg2://postgres:pw@localhost:5432/postgres"
         run: |
           pytest --cov=recordlinker --cov-report=xml tests/unit
diff --git a/pyproject.toml b/pyproject.toml
index 152434d7..268fdf57 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,7 +25,7 @@ dependencies = [
     "fhirpathpy",
     "rapidfuzz",
     # TODO: replace pyway with alembic for managing database migrations
-    "pyway"
+    "pyway==0.3.29"
 ]
 
 [project.optional-dependencies]