Fix issue with rename volume

imagr · May 9, 2019 · fe6c2e9 · fe6c2e9
1 parent 9e29fdc
commit fe6c2e9
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/Imagr/MainController.py b/Imagr/MainController.py
@@ -1991,11 +1991,9 @@ def eraseTargetVolume(self, name='Macintosh HD', format='Journaled HFS+', progre
             NSLog("Error occured when erasing volume: %@", eraseErr)
             self.errorMessage = eraseErr
         NSLog("%@", eraseOut)
-        # Reload possible targets, because '/Volumes/Macintosh HD' might not exist
-        if name != 'Macintosh HD':
-            # If the volume was renamed, or isn't named 'Macintosh HD', then we should recheck the volume list
-            self.should_update_volume_list = True
-
+        # Reload possible targets because original target name might not exist
+        self.should_update_volume_list = True
+        self.targetVolume.EnsureMountedWithRefresh()
 
     def copyLocalize_(self, item):
         if 'keyboard_layout_name' in item: