fix MigrateController
Merge pull request #537 from sergon/patch-4

fix MigrateController