Fixed the test, now it actually passes
Changed relationships to eager loading. Doesn't seem to improve anything though.