corrected the tests