diff --git a/drivers/gpu/nvgpu/gk20a/fecs_trace_gk20a.c b/drivers/gpu/nvgpu/gk20a/fecs_trace_gk20a.c
index f72b899a5..ac695823a 100644
--- a/drivers/gpu/nvgpu/gk20a/fecs_trace_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/fecs_trace_gk20a.c
@@ -77,9 +77,9 @@ static inline u64 gk20a_fecs_trace_record_ts_timestamp_v(u64 ts)
 }
 
 
-static u32 gk20a_fecs_trace_fecs_context_ptr(struct channel_gk20a *ch)
+static u32 gk20a_fecs_trace_fecs_context_ptr(struct gk20a *g, struct channel_gk20a *ch)
 {
-	return (u32) (sg_phys(ch->inst_block.sgt->sgl) >> 12LL);
+	return (u32) (gk20a_mm_inst_block_addr(g, &ch->inst_block) >> 12LL);
 }
 
 static inline int gk20a_fecs_trace_num_ts(void)
@@ -610,7 +610,7 @@ static int gk20a_fecs_trace_bind_channel(struct gk20a *g,
 	struct channel_ctx_gk20a *ch_ctx = &ch->ch_ctx;
 	struct gk20a_fecs_trace *trace = g->fecs_trace;
 	struct mem_desc *mem = &ch_ctx->gr_ctx->mem;
-	u32 context_ptr = gk20a_fecs_trace_fecs_context_ptr(ch);
+	u32 context_ptr = gk20a_fecs_trace_fecs_context_ptr(g, ch);
 	pid_t pid;
 	u32 aperture;
 
@@ -666,7 +666,7 @@ static int gk20a_fecs_trace_bind_channel(struct gk20a *g,
 
 static int gk20a_fecs_trace_unbind_channel(struct gk20a *g, struct channel_gk20a *ch)
 {
-	u32 context_ptr = gk20a_fecs_trace_fecs_context_ptr(ch);
+	u32 context_ptr = gk20a_fecs_trace_fecs_context_ptr(g, ch);
 
 	gk20a_dbg(gpu_dbg_fn|gpu_dbg_ctxsw,
 			"ch=%p context_ptr=%x", ch, context_ptr);