diff --git a/seleniumbase/console_scripts/sb_mkdir.py b/seleniumbase/console_scripts/sb_mkdir.py
index 3e13e5f8..73638e14 100755
--- a/seleniumbase/console_scripts/sb_mkdir.py
+++ b/seleniumbase/console_scripts/sb_mkdir.py
@@ -152,10 +152,11 @@ def main():
         data.append("")
         data.append("    def setUp(self):")
         data.append("        super(BaseTestCase, self).setUp()")
-        data.append("        # Add custom setUp code AFTER the super() line")
+        data.append("        # << Add custom code AFTER the super() line >>")
         data.append("")
         data.append("    def tearDown(self):")
-        data.append("        # Add custom code BEFORE the super() line")
+        data.append("        self.save_teardown_screenshot()")
+        data.append("        # << Add custom code BEFORE the super() line >>")
         data.append("        super(BaseTestCase, self).tearDown()")
         data.append("")
         data.append("    def login(self):")