diff --git a/src/modules/partition/gui/ChoicePage.cpp b/src/modules/partition/gui/ChoicePage.cpp
index 8e3e8974b..ccee2d974 100644
--- a/src/modules/partition/gui/ChoicePage.cpp
+++ b/src/modules/partition/gui/ChoicePage.cpp
@@ -451,7 +451,7 @@ ChoicePage::applyActionChoice( ChoicePage::Choice choice )
             []{},
             this );
         }
-        setNextEnabled( m_beforePartitionBarsView->selectionModel()->currentIndex().isValid() );
+        setNextEnabled( !m_beforePartitionBarsView->selectionModel()->selection().isEmpty() );
 
         connect( m_beforePartitionBarsView->selectionModel(), SIGNAL( currentRowChanged( QModelIndex, QModelIndex ) ),
                  this, SLOT( doReplaceSelectedPartition( QModelIndex, QModelIndex ) ),
@@ -492,7 +492,7 @@ ChoicePage::doReplaceSelectedPartition( const QModelIndex& current,
     } ),
     []
     {
-        setNextEnabled( m_beforePartitionBarsView->selectionModel()->currentIndex().isValid() );
+        setNextEnabled( !m_beforePartitionBarsView->selectionModel()->selection().isEmpty() );
     }, this );
 }