diff --git a/pyasic/config/__init__.py b/pyasic/config/__init__.py
index 9fb75a2a..116b57bf 100644
--- a/pyasic/config/__init__.py
+++ b/pyasic/config/__init__.py
@@ -15,8 +15,6 @@
 # ------------------------------------------------------------------------------
 from dataclasses import asdict, dataclass
 
-import toml
-
 from pyasic.config.fans import FanModeConfig
 from pyasic.config.mining import MiningModeConfig
 from pyasic.config.pools import PoolConfig
diff --git a/pyasic/config/base.py b/pyasic/config/base.py
index b082228f..3a5bfe8d 100644
--- a/pyasic/config/base.py
+++ b/pyasic/config/base.py
@@ -13,7 +13,6 @@
 #  See the License for the specific language governing permissions and         -
 #  limitations under the License.                                              -
 # ------------------------------------------------------------------------------
-from abc import ABC, abstractmethod
 from dataclasses import asdict, dataclass
 from enum import Enum
 from typing import Union
diff --git a/pyasic/config/power_scaling.py b/pyasic/config/power_scaling.py
index e776b49c..7df2fffe 100644
--- a/pyasic/config/power_scaling.py
+++ b/pyasic/config/power_scaling.py
@@ -94,7 +94,9 @@ class PowerScalingEnabled(MinerConfigValue):
     mode: str = field(init=False, default="enabled")
     power_step: int = None
     minimum_power: int = None
-    shutdown_enabled: PowerScalingShutdown = None
+    shutdown_enabled: Union[
+        PowerScalingShutdownEnabled, PowerScalingShutdownDisabled
+    ] = None
 
     @classmethod
     def from_bosminer(cls, power_scaling_conf: dict) -> "PowerScalingEnabled":