diff --git a/src/modules/partition/gui/PrettyRadioButton.cpp b/src/modules/partition/gui/PrettyRadioButton.cpp
index 4c4677cdf..08a518f1d 100644
--- a/src/modules/partition/gui/PrettyRadioButton.cpp
+++ b/src/modules/partition/gui/PrettyRadioButton.cpp
@@ -72,14 +72,14 @@ PrettyRadioButton::setIcon( const QIcon& icon )
 
 
 QSize
-PrettyRadioButton::iconSize()
+PrettyRadioButton::iconSize() const
 {
     return m_radio->iconSize();
 }
 
 
 QRadioButton*
-PrettyRadioButton::buttonWidget()
+PrettyRadioButton::buttonWidget() const
 {
     return m_radio;
 }
diff --git a/src/modules/partition/gui/PrettyRadioButton.h b/src/modules/partition/gui/PrettyRadioButton.h
index 292a68d55..ccedf25ed 100644
--- a/src/modules/partition/gui/PrettyRadioButton.h
+++ b/src/modules/partition/gui/PrettyRadioButton.h
@@ -25,20 +25,22 @@ class ClickableLabel;
 
 class PrettyRadioButton : public QWidget
 {
+    Q_OBJECT
 public:
-    PrettyRadioButton( QWidget* parent = nullptr );
+    explicit PrettyRadioButton( QWidget* parent = nullptr );
+    virtual ~PrettyRadioButton() {}
 
-    void setText( const QString& text );
+    virtual void setText( const QString& text );
 
-    void setIconSize( const QSize& size );
+    virtual void setIconSize( const QSize& size );
 
-    void setIcon( const QIcon& icon );
+    virtual void setIcon( const QIcon& icon );
 
-    QSize iconSize();
+    virtual QSize iconSize() const;
 
-    QRadioButton* buttonWidget();
+    virtual QRadioButton* buttonWidget() const;
 
-private:
+protected:
     ClickableLabel* m_label;
     QRadioButton* m_radio;
 };