diff --git a/custom_net/classifier.py b/custom_net/classifier.py
index 6bf9b09..177a1aa 100644
--- a/custom_net/classifier.py
+++ b/custom_net/classifier.py
@@ -11,7 +11,7 @@ from glob import glob
 from natsort import natsorted
 
 from .openaimodel import EncoderUNetModel, UNetModel
-from customnet_util import log_txt_as_img, default, ismap, instantiate_from_config
+from .customnet_util import log_txt_as_img, default, ismap, instantiate_from_config
 
 __models__ = {
     'class_label': EncoderUNetModel,