Update test to reflect current behavior