refactor: use environment param instead of context.environment

refactor: use environment param instead of context.environment #5702