Fix entityId bug