diff --git a/config.py b/config.py
index 373b363..c850933 100644
--- a/config.py
+++ b/config.py
@@ -34,7 +34,7 @@
 
 # Directory for training and testing dataset
 training_dir = "train2014"
-testing_dir = "test2014"
+testing_dir = "vgg/test_data"
 
 # Model, result and generated images stored path
 summary_path = "summary"
diff --git a/vgg/vgg16.py b/vgg/vgg16.py
index 1743c19..a52bbd9 100644
--- a/vgg/vgg16.py
+++ b/vgg/vgg16.py
@@ -20,7 +20,7 @@ def __init__(self, vgg16_npy_path=None):
             vgg16_npy_path = path
             print(path)
 
-        self.data_dict = np.load(vgg16_npy_path, encoding='latin1').item()
+        self.data_dict = np.load(vgg16_npy_path, encoding='latin1', allow_pickle=True).item()
         print("npy file loaded")
 
     def build(self, rgb):