1 files changed, 3 insertions, 2 deletions
diff --git a/src/modules/finished/FinishedPage.h b/src/modules/finished/FinishedPage.h
index 493c29f34..9954e6fa0 100644
--- a/src/modules/finished/FinishedPage.h
+++ b/src/modules/finished/FinishedPage.h
@@ -22,6 +22,8 @@
 
 #include <QWidget>
 
+#include "FinishedViewStep.h"
+
 namespace Ui
 {
 class FinishedPage;
@@ -33,8 +35,7 @@ class FinishedPage : public QWidget
 public:
     explicit FinishedPage( QWidget* parent = nullptr );
 
-    void setRestartNowEnabled( bool enabled );
-    void setRestartNowChecked( bool checked );
+    void setRestart( FinishedViewStep::RestartMode mode );
     void setRestartNowCommand( const QString& command );
 
     void setUpRestart();