EntitySet.get проверка наследников

Допустим у нас есть сущность в аддоне:

@JmixEntity
@Table(..,)
@Entity(name = "ts_AddOnEntity")
AddOnEntity

Где-то в аддоне есть код

AddOnEntity addOnEntity = dataManager.create(AddOnEntity.class);
SaveContext saveContext = new SaveContext();
saveContext.saving(addOnEntity);

EntitySet saved = dataManager.save(saveContext);
AddOnEntity savedAddOnEntity = saved.get(AddOnEntity.class, addOnEntity.getId());

Теперь проект использующий этот аддон решили отнаследовать сущность

@JmixEntity
@Entity
@ReplaceEntity(AddOnEntity.class)
public class ProjAddOnEntity extends AddOnEntity

Код с сохранением начинает падать. Ибо проверка класса идет ток по equals. А не по isAssignableFrom

image

Можно это подправить? Или в аддоне копировать логику? Или есть какой-то вариант для таких случаев? У EntitySet нету геттера на entities. Получается в аддоне это не учесть? И надо в проекте во всех местах где используется EntitySet.get переопределять логику?

Спасибо что сообщили о проблеме.
Завел тикет: Extended entities are missed in EntitySet methods · Issue #4757 · jmix-framework/jmix, постараемся поправить в ближайшее время.

С уважением,
Константин