changed the tests so they pass