Fix disappearance of buttons