Take care of object ref internally on QMenumodel, instead of QDBusMenuModel.

1 files changed, 4 insertions, 5 deletions

diff --git a/libqmenumodel/src/qdbusmenumodel.cpp b/libqmenumodel/src/qdbusmenumodel.cpp
index d3369dd..abf259f 100644
--- a/libqmenumodel/src/qdbusmenumodel.cpp
+++ b/libqmenumodel/src/qdbusmenumodel.cpp
@@ -89,11 +89,7 @@ void QDBusMenuModel::stop()
 /*! \internal */
 void QDBusMenuModel::serviceVanish(GDBusConnection *)
 {
-    GMenuModel *model = menuModel();
-    if (model != NULL) {
-        setMenuModel(NULL);
-        g_object_unref(model);
-    }
+    setMenuModel(NULL);
 }
 
 /*! \internal */
@@ -105,6 +101,9 @@ void QDBusMenuModel::serviceAppear(GDBusConnection *connection)
     setMenuModel(model);
     if (model == NULL) {
         stop();
+    } else {
+        //setModel take care of the ref
+        g_object_unref(model);
     }
 }