diff --git a/lib/start.js b/lib/start.js
index 1dc826c..2542225 100644
--- a/lib/start.js
+++ b/lib/start.js
@@ -1506,7 +1506,7 @@ function startRecorder(options){
                 });
                 if(continueRecord){
                     var testContent = fs.readFileSync(testFile).toString();
-                    testContent = testContent.replace(/[ \t]+function _\(str\){/, function(all){
+                    testContent = testContent.replace(/[ \t]+function _\(str\)\s*{/, function(all){
                         return arrTestCodes.join('\r\n') + '\r\n' + all;
                     });
                     fs.writeFileSync(testFile, testContent);