updating references to migrated classes