[TASK] Style: Update style on edit mode