Take env as a specific kwarg