Fixed the bug with the restart/shutdown buttons