diff --git a/drivers/staging/greybus/battery-gb.c b/drivers/staging/greybus/battery-gb.c
index 09c3290e1eaf..5d0db61d49d2 100644
--- a/drivers/staging/greybus/battery-gb.c
+++ b/drivers/staging/greybus/battery-gb.c
@@ -125,6 +125,7 @@ int gb_battery_device_init(struct gb_connection *connection)
 		return -ENOMEM;
 
 	gb->connection = connection;	// FIXME refcount!
+	connection->private = gb;
 
 	b = &gb->bat;
 	// FIXME - get a better (i.e. unique) name