Update module to support Migrate 2.6