diff --git a/.vscode/launch.json b/.vscode/launch.json
index 15a4bcc..20128e2 100644
--- a/.vscode/launch.json
+++ b/.vscode/launch.json
@@ -28,7 +28,8 @@
       "env": {
         "ERROR_UNCATCHED": "True",
         "STRICT_MODE": "True",
-        "MULTI_THREAD": "True"
+        "MULTI_THREAD": "True",
+        "CUESOR_TEST_COUNT": "3"
       }
     }
   ]
diff --git a/.vscode/tasks.json b/.vscode/tasks.json
index cfbff8d..575a290 100644
--- a/.vscode/tasks.json
+++ b/.vscode/tasks.json
@@ -9,7 +9,6 @@
           "source .venv/bin/activate;",
           "python3 tools/build.py;",
           "java -jar openapi-generator-cli.jar generate -c test/python/openapi-generator-config.yaml -g python;",
-          "python3 test/python/replace.py;",
           "python3 -m pip install ./python_generated;"
         ]
       },
@@ -19,7 +18,6 @@
           "python tools/build.py;",
           "scoop reset temurin11-jdk;",
           "java -jar openapi-generator-cli.jar generate -c test/python/openapi-generator-config.yaml -g python;",
-          "python test/python/replace.py;",
           "python -m pip install ./python_generated;"
         ]
       }
diff --git a/test/python/openapi-generator-config.yaml b/test/python/openapi-generator-config.yaml
index 29dd088..48c5e5a 100644
--- a/test/python/openapi-generator-config.yaml
+++ b/test/python/openapi-generator-config.yaml
@@ -1,4 +1,4 @@
-inputSpec: dist/compatible/openapi-3.0.yaml
+inputSpec: dist/compatible_discriminator/openapi-3.0.yaml
 outputDir: python_generated
 
 useOneOfDiscriminatorLookup: true
diff --git a/test/python/replace.py b/test/python/replace.py
deleted file mode 100644
index ab9b5f9..0000000
--- a/test/python/replace.py
+++ /dev/null
@@ -1,23 +0,0 @@
-import glob
-
-for file in glob.glob("python_generated/openapi_client/models/*.py"):
-    with open(file, "r") as f:
-        text = f.read()
-
-    indent = "    "
-
-    text = text.replace(
-        f"{indent}{indent}try:",
-        f"{indent}{indent}if match == 0:",
-    )
-    text = text.replace(
-        f"{indent}{indent}except (ValidationError, ValueError) as e:",
-        f"{indent}{indent}else:",
-    )
-    text = text.replace(
-        f"{indent}{indent}{indent}error_messages.append(str(e))",
-        f"{indent}{indent}{indent}pass",
-    )
-
-    with open(file, "w") as f:
-        f.write(text)