[BUGFIX] Do not treat all classes as entities in ClassSchema