diff --git a/src/lib/prog_loaders.c b/src/lib/prog_loaders.c
index 5048c99..183a22b 100644
--- a/src/lib/prog_loaders.c
+++ b/src/lib/prog_loaders.c
@@ -96,6 +96,8 @@
 	/* Load the cached ramstage to runtime location. */
 	stage_cache_load_stage(STAGE_RAMSTAGE, ramstage);
 
+	prog_set_arg(ramstage, cbmem_top());
+
 	if (prog_entry(ramstage) != NULL) {
 		printk(BIOS_DEBUG, "Jumping to image.\n");
 		prog_run(ramstage);
@@ -148,6 +150,9 @@
 
 	timestamp_add_now(TS_END_COPYRAM);
 
+	/* This overrides the arg fetched from the relocatable module */
+	prog_set_arg(&ramstage, cbmem_top());
+
 	prog_run(&ramstage);
 
 fail: