Update module mode during migration