diff --git a/pyasic/miners/backends/antminer.py b/pyasic/miners/backends/antminer.py
index 814c2b25..f4c39836 100644
--- a/pyasic/miners/backends/antminer.py
+++ b/pyasic/miners/backends/antminer.py
@@ -14,9 +14,9 @@
 #  limitations under the License.                                              -
 # ------------------------------------------------------------------------------
 
-from typing import List, Optional, Union
-from pathlib import Path
 import logging
+from pathlib import Path
+from typing import List, Optional, Union
 
 from pyasic.config import MinerConfig, MiningModeConfig
 from pyasic.data import AlgoHashRate, Fan, HashBoard, HashUnit
@@ -141,17 +141,24 @@ class AntminerModern(BMMiner):
             raise ValueError("File location must be provided for firmware upgrade.")
 
         try:
-            result = await self.web.update_firmware(file=file, keep_settings=keep_settings)
+            result = await self.web.update_firmware(
+                file=file, keep_settings=keep_settings
+            )
 
             if result.get("success"):
-                logging.info("Firmware upgrade process completed successfully for AntMiner.")
+                logging.info(
+                    "Firmware upgrade process completed successfully for AntMiner."
+                )
                 return "Firmware upgrade completed successfully."
             else:
                 error_message = result.get("message", "Unknown error")
                 logging.error(f"Firmware upgrade failed. Response: {error_message}")
                 return f"Firmware upgrade failed. Response: {error_message}"
         except Exception as e:
-            logging.error(f"An error occurred during the firmware upgrade process: {e}", exc_info=True)
+            logging.error(
+                f"An error occurred during the firmware upgrade process: {e}",
+                exc_info=True,
+            )
             raise
 
     async def fault_light_on(self) -> bool:
@@ -367,7 +374,7 @@ class AntminerModern(BMMiner):
 
         if web_get_conf is not None:
             try:
-                if web_get_conf["bitmain-work-mode"].isdigit():
+                if str(web_get_conf["bitmain-work-mode"]).isdigit():
                     return (
                         False if int(web_get_conf["bitmain-work-mode"]) == 1 else True
                     )