[Minor] Fix environment