Merge pull request #5 from jingweno/fix_migration_ext

Add extension to migration