diff --git a/pyHS100/__init__.py b/pyHS100/__init__.py
index 4d9c39e2..a1160082 100755
--- a/pyHS100/__init__.py
+++ b/pyHS100/__init__.py
@@ -12,11 +12,27 @@ For device type specific actions `SmartBulb` or `SmartPlug` must be used instead
 Module-specific errors are raised as `SmartDeviceException` and are expected
 to be handled by the user of the library.
 """
-# flake8: noqa
-from .smartdevice import SmartDevice, SmartDeviceException, EmeterStatus
-from .smartdevice import SmartDevice, SmartDeviceException, EmeterStatus, DeviceType
-from .smartplug import SmartPlug
-from .smartbulb import SmartBulb
-from .smartstrip import SmartStrip, SmartStripException
-from .protocol import TPLinkSmartHomeProtocol
-from .discover import Discover
+from pyHS100.discover import Discover
+from pyHS100.protocol import TPLinkSmartHomeProtocol
+from pyHS100.smartbulb import SmartBulb
+from pyHS100.smartdevice import (
+    DeviceType,
+    EmeterStatus,
+    SmartDevice,
+    SmartDeviceException,
+)
+from pyHS100.smartplug import SmartPlug
+from pyHS100.smartstrip import SmartStrip, SmartStripException
+
+__all__ = [
+    "Discover",
+    "TPLinkSmartHomeProtocol",
+    "SmartBulb",
+    "DeviceType",
+    "EmeterStatus",
+    "SmartDevice",
+    "SmartDeviceException",
+    "SmartPlug",
+    "SmartStrip",
+    "SmartStripException",
+]