1 files changed, 10 insertions, 4 deletions
diff --git a/src/arch/riscv/boot.c b/src/arch/riscv/boot.c
index 4c2be031c2..f66c02ff2e 100644
--- a/src/arch/riscv/boot.c
+++ b/src/arch/riscv/boot.c
@@ -17,11 +17,17 @@
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
-#include <arch/stages.h>
 #include <program_loading.h>
 
-void arch_payload_run(const struct payload *payload)
+void arch_prog_run(struct prog *prog)
 {
-//	uart_rx_byte(0);
-	stage_exit(prog_entry(&payload->prog));
+	void (*doit)(void *);
+
+	doit = prog_entry(prog);
+	doit(prog_entry_arg(prog));
+}
+
+void arch_payload_run(struct payload *payload)
+{
+	arch_prog_run(&payload->prog);
 }