probably betterer like this

1 files changed, 8 insertions, 12 deletions

diff --git a/src/DSi_AES.cpp b/src/DSi_AES.cpp
index 9f48519..ced8d1a 100644
--- a/src/DSi_AES.cpp
+++ b/src/DSi_AES.cpp
@@ -222,18 +222,6 @@ void WriteInputFIFO(u32 val)
     if (!(Cnt & (1<<31))) return;
 
     Update();
-
-    if (RemBlocks == 0)
-    {
-        Cnt &= ~(1<<31);
-        if (Cnt & (1<<30)) NDS::SetIRQ2(NDS::IRQ2_DSi_AES);
-        DSi::StopNDMAs(1, 0x2A);
-        DSi::StopNDMAs(1, 0x2B);
-    }
-    else
-    {
-        CheckInputDMA();
-    }
 }
 
 void CheckInputDMA()
@@ -276,6 +264,14 @@ void Update()
     }
 
     CheckOutputDMA();
+
+    if (RemBlocks == 0)
+    {
+        Cnt &= ~(1<<31);
+        if (Cnt & (1<<30)) NDS::SetIRQ2(NDS::IRQ2_DSi_AES);
+        DSi::StopNDMAs(1, 0x2A);
+        DSi::StopNDMAs(1, 0x2B);
+    }
 }