Shouldn't this be done in produce()?