Fixed the tests I accidentally overwrote