diff --git a/selfdrive/test/process_replay/update_model.py b/selfdrive/test/process_replay/update_model.py
index a53e26a19c..e0fa2adf7a 100755
--- a/selfdrive/test/process_replay/update_model.py
+++ b/selfdrive/test/process_replay/update_model.py
@@ -7,7 +7,7 @@ from selfdrive.test.process_replay.compare_logs import save_log
 from selfdrive.test.process_replay.test_processes import segments, get_segment
 from selfdrive.version import get_git_commit
 from tools.lib.logreader import LogReader
-from inject_model import inject_model
+from selfdrive.test.process_replay.inject_model import inject_model
 
 if __name__ == "__main__":