Made the test work :)