Fix missing attribute when update