diff --git a/build/tasks/output-for-loop-returns.coffee b/build/tasks/output-for-loop-returns.coffee
index 47523aa45..f9b036120 100644
--- a/build/tasks/output-for-loop-returns.coffee
+++ b/build/tasks/output-for-loop-returns.coffee
@@ -12,7 +12,7 @@ module.exports = (grunt) ->
       js = grunt.file.read(jsPath)
       method = null
       for line, index in js.split('\n')
-        [match, className, methodName] = /^\s*([a-zA-Z]+)\.prototype\.([a-zA-Z]+)\s*=\s*function\(/.exec(line) ? []
+        [match, className, methodName] = /^\s*([a-zA-Z]+)\.(?:prototype\.)?([a-zA-Z]+)\s*=\s*function\(/.exec(line) ? []
         if className and methodName
           method = "#{className}::#{methodName}"
         else