diff --git a/tests/integration/telemetry/test_telemetry.py b/tests/integration/telemetry/test_telemetry.py
index fb6efd250..c65f87489 100644
--- a/tests/integration/telemetry/test_telemetry.py
+++ b/tests/integration/telemetry/test_telemetry.py
@@ -7,6 +7,7 @@
 import time
 from uuid import uuid4
 
+import pytest
 from llama_stack_client import Agent