Use data_to_env in prepare_model_frame too