diff --git a/source/tests/jax/jax2tf/__init__.py b/source/tests/jax/jax2tf/__init__.py
index 51fa555855..755ffb04a4 100644
--- a/source/tests/jax/jax2tf/__init__.py
+++ b/source/tests/jax/jax2tf/__init__.py
@@ -1,7 +1,7 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 import tensorflow as tf
 
-from ...common import (
+from ...utils import (
     DP_TEST_TF2_ONLY,
 )