diff --git a/updater_utils.py b/updater_utils.py
index 1bcd567..209a1cd 100644
--- a/updater_utils.py
+++ b/updater_utils.py
@@ -48,7 +48,7 @@
     raise ValueError('No supported URL.')
 
 
-def replace_package(source_dir, target_dir) -> None:
+def replace_package(source_dir, target_dir, temp_file=None) -> None:
     """Invokes a shell script to prepare and update a project.
 
     Args:
@@ -59,7 +59,8 @@
     print('Updating {} using {}.'.format(target_dir, source_dir))
     script_path = os.path.join(os.path.dirname(sys.argv[0]),
                                'update_package.sh')
-    subprocess.check_call(['bash', script_path, source_dir, target_dir])
+    subprocess.check_call(['bash', script_path, source_dir, target_dir,
+                           "" if temp_file is None else temp_file])
 
 
 VERSION_SPLITTER_PATTERN: str = r'[\.\-_]'