diff --git a/src/modules/partition/gui/PartitionViewStep.cpp b/src/modules/partition/gui/PartitionViewStep.cpp
index 9ecf3b66e..5def97f51 100644
--- a/src/modules/partition/gui/PartitionViewStep.cpp
+++ b/src/modules/partition/gui/PartitionViewStep.cpp
@@ -334,7 +334,8 @@ PartitionViewStep::isAtEnd() const
 {
     if ( m_choicePage == m_widget->currentWidget() )
     {
-        if ( m_choicePage->currentChoice() == ChoicePage::Erase )
+        if ( m_choicePage->currentChoice() == ChoicePage::Erase ||
+             m_choicePage->currentChoice() == ChoicePage::Replace )
             return true;
         return false;
     }