this also means we no longer need modelenv for model_stage