diff --git a/src/main.c b/src/main.c
index 42fcf38..67e6c2e 100644
--- a/src/main.c
+++ b/src/main.c
@@ -161,7 +161,7 @@ void func_80072710(void) {
 /* 2B38 80072738 */
 void __set_vi_mode(s32 arg0) {
     switch (osTvType) {
-    case 1:
+    case OS_TV_NTSC:
         switch (arg0) {
         case 2:
             osViSetMode(&osViModeNtscLan1);
@@ -171,7 +171,8 @@ void __set_vi_mode(s32 arg0) {
             break;
         }
         break;
-    case 2:        
+        
+    case OS_TV_MPAL:        
         switch (arg0) {
         case 2:
             osViSetMode(&osViModeMpalLan1);