diff --git a/mtkclient/Library/DA/xflash/xflash_lib.py b/mtkclient/Library/DA/xflash/xflash_lib.py
index b40e621..afe9720 100755
--- a/mtkclient/Library/DA/xflash/xflash_lib.py
+++ b/mtkclient/Library/DA/xflash/xflash_lib.py
@@ -891,7 +891,7 @@ class DAXFlash(metaclass=LogBase):
             if status == 0:
                 hasflags = 0
                 # bootmode 0: shutdown 1: home screen, 2: fastboot
-                if async_mode or dl_bit or bootmode != bootmode.NORMAL:
+                if async_mode or dl_bit or bootmode != self.ShutDownModes.NORMAL:
                     hasflags = 1
                 enablewdt = 0  # Disable wdt
                 dont_resetrtc = 0  # Reset RTC