diff --git a/tests/integration/test_code.py b/tests/integration/test_code.py
index 391cf436..7963e2b7 100644
--- a/tests/integration/test_code.py
+++ b/tests/integration/test_code.py
@@ -35,7 +35,8 @@
     reason="Graham doesn't use a lustre filesystem for $HOME.",
     strict=True,
 )
-def test_check_disk_quota(
+@pytest.mark.asyncio
+async def test_check_disk_quota(
     login_node_v2: RemoteV2,
     capsys: pytest.LogCaptureFixture,
     caplog: pytest.LogCaptureFixture,
@@ -44,7 +45,7 @@ def test_check_disk_quota(
         pytest.skip(reason="Test doesn't work on localhost.")
 
     with caplog.at_level(logging.DEBUG):
-        check_disk_quota(remote=login_node_v2)
+        await check_disk_quota(remote=login_node_v2)
     # TODO: Maybe figure out a way to actually test this, (apart from just running it
     # and expecting no errors).