diff --git a/extract_nasnet_features.py b/extract_nasnet_features.py
index 019d6ad..8837188 100644
--- a/extract_nasnet_features.py
+++ b/extract_nasnet_features.py
@@ -2,7 +2,7 @@
 from keras import backend as K
 
 from utils.nasnet import NASNetMobile
-from data_loader import train_generator, val_generator
+from utils.data_loader import train_generator, val_generator
 
 sess = tf.Session()
 K.set_session(sess)
diff --git a/pretrain_nasnet_mobile.py b/pretrain_nasnet_mobile.py
index c61d416..12b5289 100644
--- a/pretrain_nasnet_mobile.py
+++ b/pretrain_nasnet_mobile.py
@@ -6,7 +6,7 @@
 from keras.optimizers import Adam
 from keras import backend as K
 
-from data_loader import features_generator
+from utils.data_loader import features_generator
 
 '''
 Below is a modification to the TensorBoard callback to perform 
diff --git a/train_mobilenet.py b/train_mobilenet.py
index 24201ad..26c751d 100644
--- a/train_mobilenet.py
+++ b/train_mobilenet.py
@@ -7,7 +7,7 @@
 from keras.optimizers import Adam
 from keras import backend as K
 
-from data_loader import train_generator, val_generator
+from utils.data_loader import train_generator, val_generator
 
 '''
 Below is a modification to the TensorBoard callback to perform 
diff --git a/train_nasnet_mobile.py b/train_nasnet_mobile.py
index af09ecb..230c96d 100644
--- a/train_nasnet_mobile.py
+++ b/train_nasnet_mobile.py
@@ -8,7 +8,7 @@
 
 from utils.nasnet import NASNetMobile
 
-from data_loader import train_generator, val_generator
+from utils.data_loader import train_generator, val_generator
 
 '''
 Below is a modification to the TensorBoard callback to perform 
diff --git a/data_loader.py b/utils/data_loader.py
similarity index 100%
rename from data_loader.py
rename to utils/data_loader.py