Fix tests calling done multiple times