2 files changed, 10 insertions, 3 deletions
diff --git a/arm9/lib/src/MI_init.c b/arm9/lib/src/MI_init.c
index 266a5f2f..4556e95a 100644
--- a/arm9/lib/src/MI_init.c
+++ b/arm9/lib/src/MI_init.c
@@ -1,11 +1,10 @@
 #include "nitro/types.h"
 #include "function_target.h"
 #include "MI_dma.h"
-
-extern void MI_SetWramBank(u32 param0);
+#include "MI_wram.h"
 
 ARM_FUNC void MI_Init(void)
 {
-    MI_SetWramBank(3);
+    MI_SetWramBank(MI_WRAM_ARM7_ALL);
     MI_StopDma(0);
 }
diff --git a/arm9/lib/src/MI_wram.c b/arm9/lib/src/MI_wram.c
new file mode 100644
index 00000000..a65bbb6f
--- /dev/null
+++ b/arm9/lib/src/MI_wram.c
@@ -0,0 +1,8 @@
+#include "function_target.h"
+#include "consts.h"
+#include "MI_wram.h"
+
+ARM_FUNC void MI_SetWramBank(MIWram cnt)
+{
+    reg_GX_VRAMCNT_WRAM = (u8)cnt;
+}