Update test to use new import path.