diff --git a/DECIMER/Efficient_Net_encoder.py b/DECIMER/Efficient_Net_encoder.py
index 2fac6e7..c6c584c 100644
--- a/DECIMER/Efficient_Net_encoder.py
+++ b/DECIMER/Efficient_Net_encoder.py
@@ -1,7 +1,6 @@
 # EfficientNet-V2 config
 import tensorflow as tf
 import DECIMER.efficientnetv2 as efficientnetv2
-from DECIMER.efficientnetv2 import effnetv2_model
 
 BATCH_SIZE_DEBUG = 2
 MODEL = "efficientnetv2-m"  # @param
@@ -17,7 +16,7 @@ def get_efficientnetv2_backbone(
             "\n...At this time we only want to use the raw "
             "(no pretraining), headless, features with no pooling ...\n"
         )
-    backbone = effnetv2_model.EffNetV2Model(model_name=model_name)
+    backbone = efficientnetv2.effnetv2_model.EffNetV2Model(model_name=model_name)
     backbone(
         tf.ones((BATCH_SIZE_DEBUG, *input_shape)), training=False, features_only=True
     )