diff --git a/dali/test/python/checkpointing/test_dali_stateless_operators.py b/dali/test/python/checkpointing/test_dali_stateless_operators.py
index 6f982fd644..dccba51af8 100644
--- a/dali/test/python/checkpointing/test_dali_stateless_operators.py
+++ b/dali/test/python/checkpointing/test_dali_stateless_operators.py
@@ -27,8 +27,7 @@
     restrict_platform,
 )
 from nose2.tools import params, cartesian_params
-from nose_utils import assert_raises, SkipTest
-from nose_utils import attr
+from nose_utils import assert_raises, SkipTest, attr
 
 # Test configuration
 batch_size = 8
diff --git a/dali/test/python/decoder/test_image.py b/dali/test/python/decoder/test_image.py
index 5fe5b3eef4..96f5dd20c8 100644
--- a/dali/test/python/decoder/test_image.py
+++ b/dali/test/python/decoder/test_image.py
@@ -23,7 +23,7 @@
 from nvidia.dali import pipeline_def
 
 from nose2.tools import params
-from nose_utils import assert_raises
+from nose_utils import assert_raises, SkipTest
 from test_utils import check_output_pattern
 from test_utils import compare_pipelines
 from test_utils import get_dali_extra_path
@@ -279,9 +279,8 @@ def check_fancy_upsampling_body(batch_size, img_type, device):
 @params(1, 8)
 def test_fancy_upsampling(batch_size):
     if nvidia.dali.backend.GetNvjpegVersion() < 12001:
-        from nose_utils import SkipTest
-
         raise SkipTest("nvJPEG doesn't support fancy upsampling in this version")
+
     data_path = os.path.join(test_data_root, good_path, "jpeg")
     compare_pipelines(
         decoder_pipe(
diff --git a/dali/test/python/decoder/test_imgcodec.py b/dali/test/python/decoder/test_imgcodec.py
index e683853cd7..fce0b853fa 100644
--- a/dali/test/python/decoder/test_imgcodec.py
+++ b/dali/test/python/decoder/test_imgcodec.py
@@ -280,9 +280,8 @@ def check_fancy_upsampling_body(batch_size, img_type, device):
 @params(1, 8)
 def test_fancy_upsampling(batch_size):
     if nvidia.dali.backend.GetNvjpegVersion() < 12001:
-        from nose_utils import SkipTest
-
         raise SkipTest("nvJPEG doesn't support fancy upsampling in this version")
+
     data_path = os.path.join(test_data_root, good_path, "jpeg")
     compare_pipelines(
         decoder_pipe(