Added typehint to migration

Signed-off-by: Daniel Bondergaard <danielboendergaard@gmail.com>
Merge pull request #1015 from danielboendergaard/migration-typehint

Added typehint to migration