small fix

2 files changed, 8 insertions, 5 deletions

diff --git a/xdg/Menu.py b/xdg/Menu.py
index 659e1b5..1618415 100644
--- a/xdg/Menu.py
+++ b/xdg/Menu.py
@@ -420,11 +420,6 @@ class MenuEntry:
 				self.DesktopEntry.write()
 			# save local one!
 			else:
-				# set self.Type
-				if self.Type == "System":
-					self.Type = "Both"
-					self.Original = MenuEntry(self.DesktopEntry.filename, self.Prefix)
-
 				path = self.__getDir()
 
 				if not os.path.isdir(os.path.dirname(os.path.join(path,self.Filename))):
diff --git a/xdg/MenuEditor.py b/xdg/MenuEditor.py
index 6821202..217e9a8 100644
--- a/xdg/MenuEditor.py
+++ b/xdg/MenuEditor.py
@@ -147,6 +147,10 @@ class MenuEditor:
 		elif nodisplay == False:
 			deskentry.set("NoDisplay", "false")
 
+		if entry.Type == "System":
+			entry.Type = "Both"
+			entry.Original = MenuEntry(entry.DesktopEntry.filename, entry.Prefix)
+
 		if len(entry.Parents) > 0:
 			sort(self.menu)
 
@@ -184,6 +188,10 @@ class MenuEditor:
 		elif nodisplay == False:
 			deskentry.set("NoDisplay", "false")
 
+		if entry.Type == "System":
+			entry.Type = "Both"
+			entry.Original = MenuEntry(entry.DesktopEntry.filename, entry.Prefix)
+
 		if isinstance(entry.Parent, Menu):
 			sort(self.menu)