Fix bug in unit tests. God, I have to test the tests now?