diff --git a/mitmproxy/tools/_main.py b/mitmproxy/tools/_main.py
index a00a3e989..0163e8d3e 100644
--- a/mitmproxy/tools/_main.py
+++ b/mitmproxy/tools/_main.py
@@ -114,7 +114,7 @@ def run(
         loop = asyncio.get_event_loop()
         for signame in ('SIGINT', 'SIGTERM'):
             try:
-                loop.add_signal_handler(getattr(signal, signame), master.shutdown)
+                loop.add_signal_handler(getattr(signal, signame), getattr(master, "prompt_for_exit", master.shutdown))
             except NotImplementedError:
                 # Not supported on Windows
                 pass