Fixed done command not working bug