fix env.py