Update sample for data load
Revert "Update sample for data load"

This reverts commit bf741df2e544564e01759ff4460a83df730e78e0.