Update environment_test.go