Made the tests work
Made the tests work