diff --git a/tests/scripts/generate_code.py b/tests/scripts/generate_code.py
index b344f8c..bc44b8c 100644
--- a/tests/scripts/generate_code.py
+++ b/tests/scripts/generate_code.py
@@ -362,9 +362,8 @@
             out.append(part)
             part = ''
         else:
+            part += str[i]
             escape = not escape and str[i] == '\\'
-            if not escape:
-                part += str[i]
     if len(part):
         out.append(part)
     return out
@@ -582,7 +581,7 @@
                                                               out_data_file.replace('\\', '\\\\')) # escape '\'
 
     # Function code
-    with open(funcs_file, 'r') as funcs_f, open(data_file, 'r') as data_f, open(out_data_file, 'w') as out_data_f:
+    with FileWrapper(funcs_file) as funcs_f, open(data_file, 'r') as data_f, open(out_data_file, 'w') as out_data_f:
         suite_deps, dispatch_code, func_code, func_info = parse_functions(funcs_f)
         snippets['functions_code'] = func_code
         snippets['dispatch_code'] = dispatch_code