diff --git a/hass_button_emulator.py b/hass_button_emulator.py
index ff10b7a..bdc3213 100755
--- a/hass_button_emulator.py
+++ b/hass_button_emulator.py
@@ -1,10 +1,9 @@
 #!/bin/python
 
+import time
+import paho.mqtt.client as mqtt
 import argparse
 import json
-import time
-
-import paho.mqtt.client as mqtt
 
 import mqtt_client
 
diff --git a/mqtt_client.py b/mqtt_client.py
index 0a2b72a..f461db4 100644
--- a/mqtt_client.py
+++ b/mqtt_client.py
@@ -16,7 +16,7 @@ class MqttClient:
         self.mixer = Mixer(mixer_ip)
         self.client = mqtt.Client(callback_api_version=mqtt.CallbackAPIVersion.VERSION2)
 
-    def mqtt_autodiscovery(self):
+    def mqtt_autodiscovery(self, client):
         j = json.dumps(
             {
                 "name": "Reset Settings",
@@ -33,7 +33,7 @@ class MqttClient:
             }
         )
 
-        self.client.publish(self.MQTT_DISCOVERY_TOPIC, payload=j, qos=0, retain=False)
+        self, client.publish(self.MQTT_DISCOVERY_TOPIC, payload=j, qos=0, retain=False)
 
     def on_connect(self, client, userdata, flags, reason_code, properties):
         print(f"Connected: {reason_code}")