diff --git a/dagfactory/dagbuilder.py b/dagfactory/dagbuilder.py
index 2d8ca7c7..c026980d 100644
--- a/dagfactory/dagbuilder.py
+++ b/dagfactory/dagbuilder.py
@@ -59,14 +59,14 @@
         from airflow.kubernetes.volume import Volume
         from airflow.kubernetes.volume_mount import VolumeMount
     else:
-        from kubernetes.client.models import V1ContainerPort as Port
         from kubernetes.client.models import (
+            V1ContainerPort as Port,
             V1EnvVar,
             V1EnvVarSource,
             V1ObjectFieldSelector,
             V1Volume,
+            V1VolumeMount as VolumeMount,
         )
-        from kubernetes.client.models import V1VolumeMount as VolumeMount
     from airflow.kubernetes.secret import Secret
     from airflow.providers.cncf.kubernetes.operators.kubernetes_pod import KubernetesPodOperator
 except ImportError: