diff --git a/tests/test_monitoring.py b/tests/test_monitoring.py
index e55afb4..bd8b148 100644
--- a/tests/test_monitoring.py
+++ b/tests/test_monitoring.py
@@ -22,7 +22,7 @@ from smolagents import (
     ToolCallingAgent,
     stream_to_gradio,
 )
-from huggingface_hub import (
+from smolagents.models import (
     ChatMessage,
     ChatMessageToolCall,
     ChatMessageToolCallDefinition,