Fix design edit

app/models/design/design.rb

@@ -11,7 +11,7 @@ class Design
   has_one :css_default, as: :css, :autosave => true, :dependent => :destroy
   has_one :layout, :autosave => true, :dependent => :destroy
   has_one :css_reset, :autosave => true, :dependent => :destroy
-  has_many :images,as: :imgs, :autosave => true, :dependent => :destroy
+  has_many :images, as: :imgs, :autosave => true, :dependent => :destroy
   has_many :javascripts, as: :js, :autosave => true, :dependent => :destroy
   has_many :pages
   has_many :themes, as: :css, :autosave => true, :dependent => :destroy

app/models/design/stylesheet.rb

@@ -7,7 +7,12 @@ class Stylesheet < DesignFile
     # self.remove_file!
     # self.remove_file_orig!
     names = {}
-    images = self.css.images
+    case self._type
+    when 'Theme'
+      images = self.design.images
+    when 'CssDefault'
+      images = self.css.images
+    end
     content.scan(/(?<=url)(.*?)(?=\))/){
       css_name = $1.gsub(' ','').gsub('(','')
       unless names.has_key?(css_name)