Changes tests to assert equals