Bug fix for milestone mark as done feature.