Update test to conform to the upgraded test module