Corrected the tests
Made the tests ok :-)