Pre segment switch

I think so. We probably should. I basically just copied the test above it.