That sounds problematic, coudn't we use an attribute or an arg for that?