Avoid calling done() too early