fixed Done button issue, I really miss JS sometimes