diff --git a/get_data.py b/get_data.py
index 3b4abe5..391bace 100644
--- a/get_data.py
+++ b/get_data.py
@@ -37,7 +37,7 @@ def send_to_mqtt(data: dict):
     publish.multiple(msgs, hostname=os.environ.get('MQTT_HOST', 'mqtt'))
 
 
-def le_advertise_packet_handler(mac: str, _, data: bytes, __):
+def le_advertise_packet_handler(mac: str, adv_type: int, data: bytes, rssi: int):
     """
     Handle new Xiaomi LYWSD03MMC BTLE advertise packet
     """
@@ -80,7 +80,9 @@ if __name__ == '__main__':
         try:
             # Called on new LE packet
             parse_le_advertising_events(
-                sock, handler=le_advertise_packet_handler, debug=False
+                sock,
+                handler=le_advertise_packet_handler,
+                debug=False,
             )
         # Scan until Ctrl-C
         except KeyboardInterrupt: