Add a workaround to load entity classes, as for some reason they stopped loading in the upgrade path test.