lib/junkie/config.rb in junkie-0.0.8 vs lib/junkie/config.rb in junkie-0.0.9

@@ -47,10 +47,13 @@
         FileUtils.mkdir_p File.dirname(Junkie::CONFIG_FILE)
         @comlete_config =
           default_config.merge_with_serialized(Junkie::CONFIG_FILE)
       end
 
-      return @comlete_config[source.class.to_s]
+      class_name = source.class.to_s
+      (class_name = source.to_s) if source.is_a? Class
+
+      return @comlete_config[class_name]
     end
 
 
     # Collects the DEFAULT_CONFIG from all including Classes
     #