diff --git a/client/client.py b/client/client.py
index 87d638f..7a5eaf1 100644
--- a/client/client.py
+++ b/client/client.py
@@ -78,8 +78,10 @@ client_id = 'mqtt-client-pimirror2'
 mqttc = mqtt.Client(mqtt.CallbackAPIVersion.VERSION2, client_id)
 mqttc.on_message = on_message
 mqttc.on_connect = on_connect
+mqttc.on_disconnect = on_disconnect
 mqttc.on_publish = on_publish
 mqttc.on_subscribe = on_subscribe
+mqttc.on_unsubscribe = on_unsubscribe
 
 # Uncomment to enable debug messages
 mqttc.on_log = on_log