diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
index f7425565..1d87a103 100644
--- a/.github/workflows/wheels.yml
+++ b/.github/workflows/wheels.yml
@@ -43,8 +43,11 @@ jobs:
 
         python setup.py bdist_wheel
 
-        # hacky until the right distribution wheel can be made...
-        find dist -name "*linux_x86_64.whl" -type f -exec bash -c 'mv "$1" "${1/linux_x86_64.whl/manylinux2014_x86_64.whl}"' _ {} \;
+        # Repair wheel (distribution support)
+        pip install auditwheel
+        auditwheel -v repair dist/* --plat manylinux2014_x86_64 --exclude libtorch_python.so --exclude libc10.so --exclude libtorch.so --exclude libtorch_cpu.so
+        rm -rf dist/*
+        mv wheelhouse/* dist/
         ls -la dist/
 
         # Run tests
diff --git a/python/tests/tools/test_commands.py b/python/tests/tools/test_commands.py
index c7aeb776..a2e08d73 100644
--- a/python/tests/tools/test_commands.py
+++ b/python/tests/tools/test_commands.py
@@ -230,7 +230,7 @@ async def test_get_or_create(self) -> None:
                         check_interval=_5_MS,
                     )
 
-                    mock_create.called_once_with(config, appdef)
+                    mock_create.assert_called_once_with(config, appdef)
                     self.assertEqual(server_info.server_handle, "slurm:///456")
                     self.assertListEqual(
                         mock_info.call_args_list,