diff --git a/drivers/gpu/nvgpu/hal/falcon/falcon_gk20a_fusa.c b/drivers/gpu/nvgpu/hal/falcon/falcon_gk20a_fusa.c
index 42cb94da3..79f561574 100644
--- a/drivers/gpu/nvgpu/hal/falcon/falcon_gk20a_fusa.c
+++ b/drivers/gpu/nvgpu/hal/falcon/falcon_gk20a_fusa.c
@@ -200,7 +200,7 @@ int gk20a_falcon_copy_to_imem(struct nvgpu_falcon *flcn, u32 dst,
 		if (i % 64U == 0U) {
 			/* tag is always 256B aligned */
 			nvgpu_writel(g,
-				base_addr + falcon_falcon_imemt_r(0),
+				base_addr + falcon_falcon_imemt_r(port),
 				tag);
 			tag++;
 		}