diff --git a/stream-pdm.c b/stream-pdm.c
index c6abab5..9083bb2 100644
--- a/stream-pdm.c
+++ b/stream-pdm.c
@@ -1,3 +1,4 @@
+#include <cc65.h>
 #include <ctype.h>
 #include <fcntl.h>
 #include <conio.h>
@@ -61,6 +62,11 @@ void main(void)
 
   videomode(VIDEOMODE_80COL);
 
+  if (doesclrscrafterexit())
+  {
+    atexit(confirm_exit);
+  }
+
   {
     int file;