Fixed the tests that failed because of me