Restore env.
small bugfix