Fix setting up the environment