diff --git a/mmc/obmc-flash-bios b/mmc/obmc-flash-bios
index c77cf8e..3fd03c2 100644
--- a/mmc/obmc-flash-bios
+++ b/mmc/obmc-flash-bios
@@ -5,6 +5,7 @@
   ro_dir="${base_dir}/running-ro"
   running_dir="${base_dir}/running"
   prsv_dir="${base_dir}/prsv"
+  staging_dir="${base_dir}/staging"
 
   if [ ! -d "${ro_dir}" ]; then
     mkdir -p "${ro_dir}"
@@ -15,6 +16,9 @@
   if [ ! -d "${prsv_dir}" ]; then
     mkdir -p "${prsv_dir}"
   fi
+  if [ ! -d "${staging_dir}" ]; then
+    mkdir -p "${staging_dir}"
+  fi
 
   # Mount the image that corresponds to the boot label as read-only to be used
   # to populate the running directory.
@@ -49,6 +53,9 @@
     cp -p ${prsv_dir}/* ${running_dir}/ 2>/dev/null || true
     rm -f ${prsv_dir}/*
 
+    # Clean up the staging dir in case of a failed update
+    rm -rf ${staging_dir}/*
+
     # Save the label
     echo "${boot_label}" > "${running_label_file}"