diff --git a/drivers/gpu/nvgpu/common/linux/vgpu/gr_vgpu.c b/drivers/gpu/nvgpu/common/linux/vgpu/gr_vgpu.c
index 1fd3e34ff..66e406478 100644
--- a/drivers/gpu/nvgpu/common/linux/vgpu/gr_vgpu.c
+++ b/drivers/gpu/nvgpu/common/linux/vgpu/gr_vgpu.c
@@ -499,8 +499,10 @@ int vgpu_gr_alloc_obj_ctx(struct channel_gk20a  *c, u32 class_num, u32 flags)
 					c->vm,
 					class_num,
 					flags);
-		if (!err)
+		if (!err) {
+			gr_ctx->tsgid = tsg->tsgid;
 			err = vgpu_gr_tsg_bind_gr_ctx(tsg);
+		}
 		if (err) {
 			nvgpu_err(g,
 				"fail to allocate TSG gr ctx buffer, err=%d", err);