Small improvement for omit new

koenbok · Aug 24, 2014 · 06a4df9 · 06a4df9
1 parent e61b373
commit 06a4df9
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/framer/Extras/OmitNew.coffee b/framer/Extras/OmitNew.coffee
@@ -1,12 +1,7 @@
 exports.enable = (module=window) ->
 
-	ClassWrapper = (Klass)->
+	ClassWrapper = (Klass) -> (args...) ->
-		CreateWrapper = (args...) ->
+		@prototype = new Klass(args...)
-			if @ is window
-				return new Klass args...
-			else
-				@prototype = new Klass args...
-		return CreateWrapper
 
 	module.Frame = ClassWrapper(Framer.Frame)
 	module.Layer = ClassWrapper(Framer.Layer)