diff --git a/lib/functions.sh b/lib/functions.sh
index 09466ff..e46c1cf 100644
--- a/lib/functions.sh
+++ b/lib/functions.sh
@@ -149,6 +149,7 @@ upgrade_federated() {
       rm /tmp/upgrade.sh
       fail "There was an error running /tmp/upgrade.sh"
     else
+      rm /tmp/upgrade.sh
       echo -ne "\nDone.\n"
     fi
   fi