diff --git a/src/driver-hidpp20.c b/src/driver-hidpp20.c
index 06896f6e..622da86f 100644
--- a/src/driver-hidpp20.c
+++ b/src/driver-hidpp20.c
@@ -466,7 +466,8 @@ hidpp20drv_probe(struct ratbag_device *device, const struct ratbag_id id)
 			goto err;
 	}
 
-	ratbag_device_init_profiles(device, 1, 8);
+	ratbag_device_init_profiles(device, 1,
+				    device->num_buttons ? device->num_buttons : 8);
 
 	return rc;
 err: