I feel like this whole test should be refactored at some point.