change the import in the sample