I updated the test. It's definitely nicer this way.