diff --git a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
index 99a6dd1b2..fd298bbf7 100644
--- a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
@@ -457,9 +457,9 @@ u32 gk20a_ce_create_context(struct gk20a *g,
 
 	/* allocate a tsg if needed */
 	ce_ctx->tsg = gk20a_tsg_open(g);
-
 	if (!ce_ctx->tsg) {
 		nvgpu_err(g, "ce: gk20a tsg not available");
+		err = -ENOMEM;
 		goto end;
 	}
 
@@ -467,6 +467,7 @@ u32 gk20a_ce_create_context(struct gk20a *g,
 	ce_ctx->ch = gk20a_open_new_channel(g, runlist_id, true);
 	if (!ce_ctx->ch) {
 		nvgpu_err(g, "ce: gk20a channel not available");
+		err = -ENOMEM;
 		goto end;
 	}
 	ce_ctx->ch->wdt_enabled = false;