No biggie, but a good refactoring step might be to pull out the shared behaviour in this step and https://github.com/alphagov/whitehall/pull/522/files#L42R24 into a feature helper.