diff --git a/tests/data/complex_example.json b/tests/data/complex_example.json
index 89a3b9324..d7d0d44c6 100644
--- a/tests/data/complex_example.json
+++ b/tests/data/complex_example.json
@@ -352,7 +352,7 @@
                                 "type": "str",
                                 "list": false
                             },
-                            "_type": "zero_shot"
+                            "_type": "ZeroShotPrompt"
                         },
                         "description": "Prompt template for Zero Shot Agent.",
                         "base_classes": [