diff --git a/source/draw.c b/source/draw.c
index 8f50ad7..1773e3f 100644
--- a/source/draw.c
+++ b/source/draw.c
@@ -155,7 +155,7 @@ void throw_error(char* error, int error_type) {
 
     switch (error_type) {
         case ERROR:
-			while (1)
+			while (aptMainLoop())
 			{
 				hidScanInput();
 				u32 kDown = hidKeysDown();
@@ -172,7 +172,7 @@ void throw_error(char* error, int error_type) {
 				}
 			}
         case WARNING:		
-			while (1)
+			while (aptMainLoop())
 				{
 					hidScanInput();
 					u32 kDown = hidKeysDown();