small ui tweak to update buttons