Entity Context missing entity in relation bug fixed