Fix small error in RestartButtonTest