diff --git a/lib/remoteproc/remoteproc.c b/lib/remoteproc/remoteproc.c
index f7f9f2df8..ad16a2d7e 100644
--- a/lib/remoteproc/remoteproc.c
+++ b/lib/remoteproc/remoteproc.c
@@ -847,6 +847,7 @@ int remoteproc_load_noblock(struct remoteproc *rproc,
 			}
 			rproc->rsc_table = rsc_table;
 			rproc->rsc_len = rsc_size;
+			rproc->rsc_io = *io;
 			metal_free_memory(lrsc_table);
 		}