diff --git a/tests/core/test_run_config.py b/tests/core/test_run_config.py
index b09cdac5..9de52d13 100644
--- a/tests/core/test_run_config.py
+++ b/tests/core/test_run_config.py
@@ -5,8 +5,8 @@ import pytest
 from cai.sdk.agents import Agent, RunConfig, Runner
 from cai.sdk.agents.models.interface import Model, ModelProvider
 
-from .fake_model import FakeModel
-from .test_responses import get_text_message
+from tests.fake_model import FakeModel
+from tests.core.test_responses import get_text_message
 
 
 class DummyProvider(ModelProvider):
diff --git a/tests/core/test_run_step_execution.py b/tests/core/test_run_step_execution.py
index 7e0fee35..32dd6c88 100644
--- a/tests/core/test_run_step_execution.py
+++ b/tests/core/test_run_step_execution.py
@@ -27,7 +27,7 @@ from cai.sdk.agents._run_impl import (
     SingleStepResult,
 )
 
-from .test_responses import (
+from tests.core.test_responses import (
     get_final_output_message,
     get_function_tool,
     get_function_tool_call,
diff --git a/tests/core/test_run_step_processing.py b/tests/core/test_run_step_processing.py
index cce07474..a80b7134 100644
--- a/tests/core/test_run_step_processing.py
+++ b/tests/core/test_run_step_processing.py
@@ -25,7 +25,7 @@ from cai.sdk.agents import (
 )
 from cai.sdk.agents._run_impl import RunImpl
 
-from .test_responses import (
+from tests.core.test_responses import (
     get_final_output_message,
     get_function_tool,
     get_function_tool_call,
diff --git a/tests/others/test_function_schema.py b/tests/others/test_function_schema.py
index a68519be..9f184d72 100644
--- a/tests/others/test_function_schema.py
+++ b/tests/others/test_function_schema.py
@@ -99,7 +99,7 @@ def varargs_function(x: int, *numbers: float, flag: bool = False, **kwargs: Any)
 def test_varargs_function():
     """Test a function that uses *args and **kwargs."""
 
-    func_schema = function_schema(varargs_function)
+    func_schema = function_schema(varargs_function, strict_json_schema=False)
     # Check JSON schema structure
     assert isinstance(func_schema.params_json_schema, dict)
     assert func_schema.params_json_schema.get("title") == "varargs_function_args"