diff --git a/kano-wand-2/wand.py b/kano-wand-2/wand.py
index 2a77bda..6b410d7 100644
--- a/kano-wand-2/wand.py
+++ b/kano-wand-2/wand.py
@@ -346,7 +346,7 @@ class Wand(object):
             print("Subscribing to button notification")
 
         self._button_subscribed = True
-        await self._dev.start_notify(IO.USER_BUTTON_CHAR.value, self._on_button)
+        await self._dev.start_notify(IO.USER_BUTTON_CHAR.value, self.handle_notification)
         # with self._lock:
         # if not hasattr(self, "_button_handle"):
         #     handle = self._io_service.getCharacteristics(IO.USER_BUTTON_CHAR.value)[0]
@@ -591,18 +591,18 @@ class Wand(object):
         pass
 
 
-    def handleNotification(self, cHandle, data):
+    def handle_notification(self, sender, data):
         """Handle notifications subscribed to
 
         Arguments:
             cHandle {int} -- Handle of notification
             data {bytes} -- Data from device
         """
-        if cHandle == SENSOR.QUATERNIONS_CHAR.value:
+        if sender == SENSOR.QUATERNIONS_CHAR.value:
             self._on_position(data)
-        elif cHandle == IO.USER_BUTTON_CHAR.value:
+        elif sender == IO.USER_BUTTON_CHAR.value:
             self._on_button(data)
-        elif cHandle == SENSOR.TEMP_CHAR.value:
+        elif sender == SENSOR.TEMP_CHAR.value:
             self._on_temperature(data)
-        elif cHandle == IO.BATTERY_CHAR.value:
+        elif sender == IO.BATTERY_CHAR.value:
             self._on_battery(data)