Fixes UI show even if button is toggled