Fix root partition preparation