diff --git a/tests/integ/test_airflow_config.py b/tests/integ/test_airflow_config.py
index df0311eca3..eeef7643ae 100644
--- a/tests/integ/test_airflow_config.py
+++ b/tests/integ/test_airflow_config.py
@@ -42,7 +42,6 @@
 from sagemaker.sklearn import SKLearn
 from sagemaker.tensorflow import TensorFlow
 from sagemaker.utils import sagemaker_timestamp
-from sagemaker.workflow import airflow as sm_airflow
 from sagemaker.xgboost import XGBoost
 from tests.integ import datasets, DATA_DIR
 from tests.integ.record_set import prepare_record_set_from_local_files
@@ -54,7 +53,8 @@
     seconds_to_sleep=6,
 ):
     try:
-        from airflow import utils
+        import sagemaker.workflow.airflow as sm_airflow
+        import airflow.utils as utils
         from airflow import DAG
         from airflow.providers.amazon.aws.operators.sagemaker import SageMakerTrainingOperator
         from airflow.providers.amazon.aws.operators.sagemaker_transform import (
@@ -64,6 +64,11 @@
         break
     except ParsingError:
         pass
+    except ValueError as ve:
+        if "Unable to configure formatter" in str(ve):
+            print(f"Received: {ve}")
+        else:
+            raise ve
 
 PYTORCH_MNIST_DIR = os.path.join(DATA_DIR, "pytorch_mnist")
 PYTORCH_MNIST_SCRIPT = os.path.join(PYTORCH_MNIST_DIR, "mnist.py")