diff --git a/tests/unit/aiplatform/test_logdir_loader.py b/tests/unit/aiplatform/test_logdir_loader.py
index c2e22a2b19..44a66a79d8 100644
--- a/tests/unit/aiplatform/test_logdir_loader.py
+++ b/tests/unit/aiplatform/test_logdir_loader.py
@@ -21,15 +21,14 @@
 import shutil
 import tempfile
 
-import sys
-
 import pytest
 
-if sys.version_info >= (3, 13):
-    pytest.skip("Tensorflow not available for Python 3.13+", allow_module_level=True)
+try:
+    import tensorflow as tf
+except ImportError:
+    pytest.skip("Tensorflow not available", allow_module_level=True)
 
 from google.cloud.aiplatform.tensorboard import logdir_loader
-import tensorflow as tf
 
 from tensorboard.backend.event_processing import directory_loader
 from tensorboard.backend.event_processing import event_file_loader
diff --git a/tests/unit/aiplatform/test_uploader_utils.py b/tests/unit/aiplatform/test_uploader_utils.py
index 6b11986aa8..5b4356ebc0 100644
--- a/tests/unit/aiplatform/test_uploader_utils.py
+++ b/tests/unit/aiplatform/test_uploader_utils.py
@@ -20,14 +20,12 @@
 from unittest import mock
 
 from absl.testing import absltest
-import sys
-
 import pytest
 
-if sys.version_info >= (3, 13):
-    pytest.skip("Tensorboard not available for Python 3.13+", allow_module_level=True)
-
-from google.cloud.aiplatform.tensorboard import uploader_utils
+try:
+    from google.cloud.aiplatform.tensorboard import uploader_utils
+except ImportError:
+    pytest.skip("Tensorboard not available", allow_module_level=True)
 
 
 class FakeTime: