diff --git a/stylenet.py b/stylenet.py
index 4d3485b..76c74a4 100644
--- a/stylenet.py
+++ b/stylenet.py
@@ -4,7 +4,7 @@
 
 
 
-_stylenet_config_keys_ = [
+_stylenet_config_keys_ = [ 'type',
     'size', 'content_image', 'style_images', 
     'style_image', 'content_masks', 'style_blend_weights', 
     'num_octaves', 'style_scale', 'num_iterations',