From 5ca32cdac77bec4ef8a0545c86c1956f02bd8cf9 Mon Sep 17 00:00:00 2001
From: Felipebmaitan <felipebmaitan@protonmail.com>
Date: Fri, 23 Oct 2026 13:14:44 -0300
Subject: [PATCH] test(job-manager): align reference entry helper calls

---
 tools/run_job_manager_reference.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tools/run_job_manager_reference.py b/tools/run_job_manager_reference.py
index 153b87f38b..2b1246c798 100644
--- a/tools/run_job_manager_reference.py
+++ b/tools/run_job_manager_reference.py
@@ -3,6 +3,7 @@ import hashlib
 import json
 import os
 import shutil
+import struct
 import subprocess
 import sys
 import tempfile
@@ -61,7 +62,9 @@ def run(args, directory):
         else:
             unwrap(args.module, target)
     source = directory / "input.elf"
-    source.write_bytes(executable("jobref.prx", symbol="Run_nid_no_patch"))
+    entry = executable("jobref.prx", symbol="Run_nid_no_patch")
+    entry[0x4000:0x400f] = b"\x48\x83\xec\x08\xff\x15" + struct.pack("<i", 0x4a20 - 0x400a) + b"\x48\x83\xc4\x08\xc3"
+    source.write_bytes(entry)
     output = directory / f"reference{suffix}"
     options = ["--skip-sce-module"] if args.host else []
     platform = ["--windows"] if sys.platform == "win32" else []