diff --git a/DenseNet121-ImageNet b/DenseNet121-ImageNet
index c550008..4eaf5bf 100644
--- a/DenseNet121-ImageNet
+++ b/DenseNet121-ImageNet
@@ -5,7 +5,7 @@ import os
 os.environ['TL_BACKEND'] = 'tensorflow'
 
 import tensorlayer as tl
-from densenet import densenet
+from .densenet import densenet
 
 tl.logging.set_verbosity(tl.logging.DEBUG)
 def load_ImageNet_dataset(shape=(-1, 256, 256, 3), plotable=False):