diff --git a/pydra/engine/tests/test_audit.py b/pydra/engine/tests/test_audit.py
index a4ac3c7ac..52f09f88b 100644
--- a/pydra/engine/tests/test_audit.py
+++ b/pydra/engine/tests/test_audit.py
@@ -302,3 +302,15 @@ def TestFunc(a: int, b: float = 0.1):
     # commented out to speed up testing
     collect_messages(tmpdir / funky._checksum, message_path, ld_op="compact")
     assert (tmpdir / funky._checksum / "messages.jsonld").exists()
+
+
+def test_audit_resource():
+
+    @python.define(outputs=["out"])
+    def AddVar(a: int, b: int) -> int:
+        return a + b
+
+    task0 = AddVar(a=4, b=5)
+    outputs = task0(audit_flags=AuditFlag.RESOURCE)
+
+    print(outputs)
diff --git a/pyproject.toml b/pyproject.toml
index e93f75dc9..2579b9f5e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,6 +14,7 @@ dependencies = [
     "filelock >=3.0.0",
     "fileformats >=0.15.0a7",
     "platformdirs >=2",
+    "psutil >=5.9.0",
 ]
 license = { file = "LICENSE" }
 authors = [{ name = "Nipype developers", email = "neuroimaging@python.org" }]
@@ -78,7 +79,6 @@ test = [
     "fileformats-extras >=0.15.0a7",
     "numpy >=1.26",
     "pyld >=2.0",
-    "psutil >=5.9.0",
     "python-dateutil >=2.8.2",
     "tornado >=6.1",
 ]