Fixed done button redirect bug