diff --git a/plugins/device/client.go b/plugins/device/client.go
index 4548689cb..d20146e75 100644
--- a/plugins/device/client.go
+++ b/plugins/device/client.go
@@ -81,7 +81,7 @@ func (d *devicePluginClient) Reserve(deviceIDs []string) (*ContainerReservation,
 	}
 
 	// Make the request
-	resp, err := d.client.Reserve(context.Background(), req)
+	resp, err := d.client.Reserve(d.doneCtx, req)
 	if err != nil {
 		return nil, err
 	}