Fixed done button not working