Fixed failing test to make correct assertion.