lib/traits/damaging.rb in gosu_extensions-0.3.5 vs lib/traits/damaging.rb in gosu_extensions-0.3.6

@@ -44,20 +44,18 @@
   
   def self.included klass
     klass.extend ClassMethods
   end
   
-  def initialize window
+  def initialize *args
     raise DamageMissingError.new unless respond_to?(:damage)
-    super window
+    super *args
   end
   
   module ClassMethods
-    
     def damage damage = nil, &block
-      block = lambda { damage } unless block_given?
+      block ||= lambda { damage }
       define_method :damage, &block
     end
-    
   end
   
 end
\ No newline at end of file