1 files changed, 2 insertions, 2 deletions
diff --git a/packages/meshbay-node/tests/test_audit.py b/packages/meshbay-node/tests/test_audit.py
index b5b98b5..1a685ff 100644
--- a/packages/meshbay-node/tests/test_audit.py
+++ b/packages/meshbay-node/tests/test_audit.py
@@ -1,13 +1,13 @@
 """Tests for the node audit store (legal compliance IP/action logging)."""
 
 import asyncio
-import inspect
 import time
 
 import pytest
 from meshbay_node.audit import AuditStore
 from meshbay_node.config import Config, HubConfig, KeystoreConfig, NodeConfig
 from meshbay_node.daemon import NodeDaemon
+from node_source import daemon_class_source
 
 
 @pytest.fixture
@@ -132,5 +132,5 @@ async def test_daemon_purges_the_audit_log(audit, tmp_path):
 
 
 def test_daemon_starts_the_purge():
-    source = inspect.getsource(NodeDaemon)
+    source = daemon_class_source()
     assert "create_task(self._purge_audit_log())" in source