diff --git a/scripts/generate_test_code.py b/scripts/generate_test_code.py
index 2ac00add2c..4b7f9397e9 100755
--- a/scripts/generate_test_code.py
+++ b/scripts/generate_test_code.py
@@ -762,7 +762,7 @@ def escaped_split(inp_str, split_char):
         raise ValueError('Expected split character. Found string!')
     out = re.sub(r'(\\.)|' + split_char,
                  lambda m: m.group(1) or '\n', inp_str,
-                 len(inp_str)).split('\n')
+                 count=len(inp_str)).split('\n')
     out = [x for x in out if x]
     return out