Removed load function off the migration set.

It was only called internally and added a level of callbacks between migrate() and load() which wasn't necessary.