Update test to account for the new argument