Fixed markAsDone case