Add entityType to context