Fixed failing test to make a more general assertion.