Restore menu highlight after changing themes