migration: Add support for modules