Fix missing attribute