diff --git a/tests/test_agents.py b/tests/test_agents.py
index 5486b2d..06d9554 100644
--- a/tests/test_agents.py
+++ b/tests/test_agents.py
@@ -360,7 +360,7 @@ class AgentTests(unittest.TestCase):
         with console.capture() as capture:
             agent.run("Count to 3")
         str_output = capture.get()
-        assert "import under additional_authorized_imports" in str_output
+        assert "import under `additional_authorized_imports`" in str_output
 
     def test_multiagents(self):
         class FakeModelMultiagentsManagerAgent: