lib/jaspion/kilza/class.rb in jaspion-kilza-1.1.0 vs lib/jaspion/kilza/class.rb in jaspion-kilza-1.1.1

@@ -14,11 +14,12 @@
 
       # Initializes a Class object
       #
       # @param name [String] Class Name
       def initialize(name)
-        @name = Kilza.normalize(name).capitalize
+        @name = Kilza.clean(name)
+        @name[0] = @name[0].capitalize
         @properties = []
         @imports = []
       end
 
       # Adds a new property
@@ -51,10 +52,10 @@
         path = File.join(erb_path, file_name + '.erb')
         eruby = Erubis::Eruby.new(File.read(path))
 
         s = Kilza::Source.new
         s.source = eruby.result(binding)
-        s.file_name = @name.capitalize + '.' + file_name
+        s.file_name = @name + '.' + file_name
         s
       end
 
       def to_s
         properties = []