diff --git a/serving/sf2f/models/model_setup.py b/serving/sf2f/models/model_setup.py
index e077e47..8f31612 100644
--- a/serving/sf2f/models/model_setup.py
+++ b/serving/sf2f/models/model_setup.py
@@ -4,7 +4,10 @@
 except:
     from ..models import *
 from copy import deepcopy
-from utils import update_values, load_model_state
+try:
+    from ..utils import update_values, load_model_state
+except:
+    from .utils import update_values, load_model_state
 import glog as log
 
 def build_model(opts, image_size, checkpoint_start_from=None):