diff --git a/.github/workflows/build-publish.yml b/.github/workflows/build-publish.yml
index e8bc0582..737c653e 100644
--- a/.github/workflows/build-publish.yml
+++ b/.github/workflows/build-publish.yml
@@ -40,9 +40,9 @@ jobs:
       with:
         python-version: '3.x'
     - name: Install dependencies
-      run: python -m pip install --upgrade setuptools wheel
+      run: python -m pip install --upgrade setuptools wheel build
     - name: Build wheels
-      run: python setup.py sdist bdist_wheel
+      run: python -m build --sdist --wheel
     - name: Store wheels
       uses: actions/upload-artifact@v2
       with:
diff --git a/tests/test_dm_control.py b/tests/test_dm_control.py
index a469e774..8dfc599f 100644
--- a/tests/test_dm_control.py
+++ b/tests/test_dm_control.py
@@ -55,6 +55,7 @@ def test_dm_control_suite_envs():
         "It seems a Box observation space is an image but the upper and lower bounds are not in [0, 255]. Generally, CNN policies assume observations are within that range, so you may encounter an issue if the observation values are not.",
         "arrays to stack must be passed as a 'sequence' type such as list or tuple. Support for non-sequence iterables such as generators is deprecated as of NumPy 1.16 and will raise an error in the future.",
         "Calling `env.close()` on the closed environment should be allowed, but it raised an exception: _data",
+        "Calling `env.close()` on the closed environment should be allowed, but it raised an exception: 'Physics' object has no attribute '_data'",
     ]
 ]
 CHECK_ENV_IGNORE_WARNINGS.append(