diff --git a/releasetools.py b/releasetools.py
index 35656652f5c8c49b9ec9c65d7674770590b29d13..2eb5208ce801e50f7b4d89acb84b690c6562ab21 100644
--- a/releasetools.py
+++ b/releasetools.py
@@ -32,4 +32,6 @@ def AddImage(info, basename, dest):
 def OTA_InstallEnd(info):
   info.script.Print("Patching firmware images...")
   AddImage(info, "dtbo.img", "/dev/block/bootdevice/by-name/dtbo")
+  AddImage(info, "vbmeta.img", "/dev/block/bootdevice/by-name/vbmeta")
+  AddImage(info, "vbmeta_system.img", "/dev/block/bootdevice/by-name/vbmeta_system")
   return