Fixed the done button