Remove check for model.state='DONE'