lib/nilify_blanks.rb in nilify_blanks-1.0.0 vs lib/nilify_blanks.rb in nilify_blanks-1.0.1

@@ -20,11 +20,14 @@
         options[:except] = options[:except].map(&:to_s)
       end
       
       cattr_accessor :nilify_blanks_columns
       
-      self.nilify_blanks_columns = self.content_columns.reject {|c| !c.null }.map {|c| c.name.to_s }
-      self.nilify_blanks_columns = self.nilify_blanks_columns.select {|c| options[:only].include?(c) } if options[:only]
+      if options[:only]
+        self.nilify_blanks_columns = options[:only].clone
+      else
+        self.nilify_blanks_columns = self.content_columns.reject {|c| !c.null }.map {|c| c.name.to_s }
+      end
       self.nilify_blanks_columns -= options[:except] if options[:except]
       self.nilify_blanks_columns = self.nilify_blanks_columns.map(&:to_s)
       
       options[:before] ||= :save
       class_eval "before_#{options[:before]} :nilify_blanks"