diff --git a/releasetools.py b/releasetools.py
index 2692a00..a4611a2 100644
--- a/releasetools.py
+++ b/releasetools.py
@@ -23,7 +23,7 @@ def FullOTA_InstallEnd(info):
   return
 
 def IncrementalOTA_InstallEnd(info):
-  input_zip = info.input_zip
+  input_zip = info.target_zip
   OTA_InstallEnd(info, input_zip)
   return