Fix done button