fixed the tests to search for correct type