A little more fixing for Mark Done button