Actually fix unit test properly, not derpily.