Fix to migration
Added some comments