diff --git a/releasetools/releasetools.py b/releasetools/releasetools.py
index 135ab30..e5441de 100644
--- a/releasetools/releasetools.py
+++ b/releasetools/releasetools.py
@@ -22,6 +22,7 @@
   return
 
 def IncrementalOTA_InstallEnd(info):
+  info.input_zip = info.target_zip
   OTA_InstallEnd(info)
   return