diff --git a/recipes-diya/initramfs/files/diya-update b/recipes-diya/initramfs/files/diya-update
index 08ed1fa..ead9210 100755
--- a/recipes-diya/initramfs/files/diya-update
+++ b/recipes-diya/initramfs/files/diya-update
@@ -54,6 +54,7 @@ rootfs()
         return 1
     fi
     sync
+    cd
     echo "Unmount the rootfs partition"
     umount $TMP_MOUNT
     echo "Done"