Fix done/not done display bug