diff --git a/tests/unit/core/test_version.py b/tests/unit/core/test_version.py
index e93ba531f..ca7bfcf8b 100644
--- a/tests/unit/core/test_version.py
+++ b/tests/unit/core/test_version.py
@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+import pytest
 from packaging.version import Version
 
 import merlin.core