fixing custom env
fixing custom env