Update test to work with new helper