diff --git a/src/modules/partition/core/PartitionCoreModule.cpp b/src/modules/partition/core/PartitionCoreModule.cpp
index 98d7b6c21..c281d30db 100644
--- a/src/modules/partition/core/PartitionCoreModule.cpp
+++ b/src/modules/partition/core/PartitionCoreModule.cpp
@@ -510,6 +510,8 @@ PartitionCoreModule::revertDevice( Device* dev )
     CoreBackend* backend = CoreBackendManager::self()->backend();
     Device *newDev = backend->scanDevice( devInfo->device->deviceNode() );
     devInfo->device.reset( newDev );
+    devInfo->partitionModel->init( newDev, m_osproberLines );
+
     m_deviceModel->swapDevice( dev, newDev );
 
     QList< Device* > devices;
@@ -518,8 +520,6 @@ PartitionCoreModule::revertDevice( Device* dev )
 
     m_bootLoaderModel->init( devices );
 
-    devInfo->partitionModel->init( newDev, m_osproberLines );
-
     updateIsDirty();
 }