Fixing the restart button on_press event.