diff --git a/doodle.py b/doodle.py
index abe9f6a..6e2fe03 100644
--- a/doodle.py
+++ b/doodle.py
@@ -233,6 +233,10 @@ def __init__(self):
             self.content_map_original = np.zeros(self.content_img_original.shape[:2]+(1,))
             self.semantic_weight = 0.0
 
+        if self.style_map_original is None:
+            self.style_map_original = np.zeros(self.style_img_original.shape[:2]+(1,))
+            self.semantic_weight = 0.0
+
         if self.content_img_original is None:
             self.content_img_original = np.zeros(self.content_map_original.shape[:2]+(3,))
             args.content_weight = 0.0