diff --git a/pkg/action/upgrade.go b/pkg/action/upgrade.go
index 58171cd..7810211 100644
--- a/pkg/action/upgrade.go
+++ b/pkg/action/upgrade.go
@@ -176,8 +176,9 @@ func (u *UpgradeAction) Run() (err error) {
 			umount, err = e.MountRWPartition(persistentPart)
 			if err != nil {
 				u.config.Logger.Warnf("could not mount persistent partition: %s", err.Error())
+			} else {
+				cleanup.Push(umount)
 			}
-			cleanup.Push(umount)
 		}
 	}