From 6739b81c2ac119636f816b2b917dc4e372bfb0df Mon Sep 17 00:00:00 2001
From: Adriaan de Groot <groot@kde.org>
Date: Wed, 16 May 2018 07:13:35 -0400
Subject: [PATCH] [partition] Tell PartitionInfo about desired flags

 - When creating a partition, or changing flags, tell
   the PartitionInfo about those desired flags.
---
 src/modules/partition/core/PartitionCoreModule.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/modules/partition/core/PartitionCoreModule.cpp b/src/modules/partition/core/PartitionCoreModule.cpp
index 43ba33d7b..f636d15bb 100644
--- a/src/modules/partition/core/PartitionCoreModule.cpp
+++ b/src/modules/partition/core/PartitionCoreModule.cpp
@@ -258,6 +258,7 @@ PartitionCoreModule::createPartition( Device* device,
     {
         SetPartFlagsJob* fJob = new SetPartFlagsJob( device, partition, flags );
         deviceInfo->jobs << Calamares::job_ptr( fJob );
+        PartitionInfo::setFlags( partition, flags );
     }
 
     refresh();
@@ -381,8 +382,8 @@ PartitionCoreModule::setPartitionFlags( Device* device,
     PartitionModel::ResetHelper( partitionModelForDevice( device ) );
 
     SetPartFlagsJob* job = new SetPartFlagsJob( device, partition, flags );
-
     deviceInfo->jobs << Calamares::job_ptr( job );
+    PartitionInfo::setFlags( partition, flags );
 
     refresh();
 }