diff --git a/video_prediction_tools/model_modules/video_prediction/ops.py b/video_prediction_tools/model_modules/video_prediction/ops.py
index 1d2e9f2eac608d2a9ec61e24a354882b3acce2de..7e41af95c2941f610235c7fa5744d326d333418d 100644
--- a/video_prediction_tools/model_modules/video_prediction/ops.py
+++ b/video_prediction_tools/model_modules/video_prediction/ops.py
@@ -1065,7 +1065,7 @@ def get_norm_layer(layer_type):
     elif layer_type == 'layer':
         layer = tf.contrib.layers.layer_norm
     elif layer_type == 'instance':
-        from video_prediction.layers import fused_instance_norm
+        from model_modules.video_prediction.layers import fused_instance_norm
         layer = fused_instance_norm
     elif layer_type == 'none':
         layer = tf.identity