Fixed the custom environment in the Playground