diff --git a/misc/vm-core.func b/misc/vm-core.func
index f283879..28fec13 100644
--- a/misc/vm-core.func
+++ b/misc/vm-core.func
@@ -346,8 +346,6 @@ cleanup() {
   if [[ "$(dirs -p | wc -l)" -gt 1 ]]; then
     popd >/dev/null || true
   fi
-  post_update_to_api "done" "none"
-  [[ -n "${TEMP_DIR:-}" && -d "$TEMP_DIR" ]] && rm -rf "$TEMP_DIR"
 }
 
 check_root() {
diff --git a/vm/debian-vm.sh b/vm/debian-vm.sh
index 8e20fe1..c5e0492 100644
--- a/vm/debian-vm.sh
+++ b/vm/debian-vm.sh
@@ -47,6 +47,8 @@ function error_handler() {
 get_valid_nextid
 cleanup_vmid
 cleanup
+post_update_to_api "done" "none"
+[[ -n "${TEMP_DIR:-}" && -d "$TEMP_DIR" ]] && rm -rf "$TEMP_DIR"
 check_root
 pve_check
 arch_check