Update test to reflect proper usage