diff --git a/hassio/dock/supervisor.py b/hassio/dock/supervisor.py
index 18a900fce..41c463338 100644
--- a/hassio/dock/supervisor.py
+++ b/hassio/dock/supervisor.py
@@ -1,9 +1,13 @@
 """Init file for HassIO docker object."""
+import logging
 import os
 
 import docker
 
 from . import DockerBase
+from ..const import RESTART_EXIT_CODE
+
+_LOGGER = logging.getLogger(__name__)
 
 
 class DockerSupervisor(DockerBase):