diff --git a/extractor.py b/extractor.py
index d9ca868b1f31c989b4bb7baf0af1d5332948742f..a3d8cd804e1d685f83bf112292ad279cdc7ed679 100644
--- a/extractor.py
+++ b/extractor.py
@@ -47,7 +47,7 @@ class Extractor():
 
         if self.weights is None:
             # For imagenet/default network:
-            features = features[0][0][0]
+            features = features[0]
         else:
             # For loaded network:
             features = features[0]