[fix] Fix `env_set`