Only set specific entity type