Reset the partition model before swapping the new Device into the DM.

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();
 }