Add missing dependancy in migration.