diff --git a/model_server/tests/core/test_function_calling.py b/model_server/tests/core/test_function_calling.py
index 9f1d0750..5ad575cd 100644
--- a/model_server/tests/core/test_function_calling.py
+++ b/model_server/tests/core/test_function_calling.py
@@ -1,5 +1,5 @@
 import pytest
-
+import time
 from src.commons.globals import handler_map
 from src.core.utils.model_utils import ChatMessage, Message