Update test to match code.