changed the logic, this should fix the tests