1 files changed, 5 insertions, 2 deletions
diff --git a/tests/units/result_manager/test_models.py b/tests/units/result_manager/test_models.py
index bc7ba8a..2276153 100644
--- a/tests/units/result_manager/test_models.py
+++ b/tests/units/result_manager/test_models.py
@@ -2,18 +2,21 @@
 # Use of this source code is governed by the Apache License 2.0
 # that can be found in the LICENSE file.
 """ANTA Result Manager models unit tests."""
+
 from __future__ import annotations
 
-from typing import Any, Callable
+from typing import TYPE_CHECKING, Any, Callable
 
 import pytest
 
 # Import as Result to avoid pytest collection
-from anta.result_manager.models import TestResult as Result
 from tests.data.json_data import TEST_RESULT_SET_STATUS
 from tests.lib.fixture import DEVICE_NAME
 from tests.lib.utils import generate_test_ids_dict
 
+if TYPE_CHECKING:
+    from anta.result_manager.models import TestResult as Result
+
 
 class TestTestResultModels:
     """Test components of anta.result_manager.models."""