fix(ui): fix initilisation logic