Implement swappable model support for migrations