fixing incorrect usage of plot_env_init in tests