diff --git a/httprunner/templates/locustfile_template b/httprunner/templates/locustfile_template
index f5f481c5..1b61b4be 100644
--- a/httprunner/templates/locustfile_template
+++ b/httprunner/templates/locustfile_template
@@ -23,7 +23,7 @@ class WebPageTasks(TaskSet):
         for teststep in teststeps:
             try:
                 self.test_runner.run_test(teststep)
-            except (MyBaseError, MyBaseFailure) as ex:
+            except (AssertionError, MyBaseError, MyBaseFailure) as ex:
                 request_failure.fire(
                     request_type=teststep.get("request", {}).get("method"),
                     name=teststep.get("name"),