diff --git a/src/main.c b/src/main.c
index da68494..ff67ab4 100644
--- a/src/main.c
+++ b/src/main.c
@@ -182,8 +182,10 @@ asm (
   "lda #$80     \n"  // DMA_CTRL_START
   //"tsb $DF00    \n"  // DMA_CTRL
   ".byte $0C,$00,$DF\n"
-"wait:   lda $DF01    \n"  // DMA_STATUS
-  "bmi wait     \n"  // Wait for DMA
+//"wait_start:   lda $DF01    \n"  // DMA_STATUS
+//	"bpl wait_start         \n"  // Wait for DMA
+"wait_end:   lda $DF01    \n"  // DMA_STATUS
+  "bmi wait_end     \n"  // Wait for DMA
  //"stz $DF00    \n"  // DMA Off
   ".byte $9c,$00,$DF\n"
   "nop          \n"
@@ -669,6 +671,7 @@ void runSimulation(void) {
 		a23d2Draw();
 #else
 
+		graphicsWaitVerticalBlank();
 		dmaClear(pageZero ? 0x6c000 : 0x58000, 0x12c00, 0);
 
 #endif