diff --git a/src/kdescendantsproxymodel.cpp b/src/kdescendantsproxymodel.cpp
index bcce595..79fe299 100644
--- a/src/kdescendantsproxymodel.cpp
+++ b/src/kdescendantsproxymodel.cpp
@@ -279,9 +279,9 @@ void KDescendantsProxyModel::setSourceModel(QAbstractItemModel *_sourceModel)
       SLOT(sourceModelDestroyed())
     };
 
-    if (_sourceModel) {
+    if (sourceModel()) {
         for (int i = 0; i < int(sizeof modelSignals / sizeof *modelSignals); ++i)
-            disconnect(_sourceModel, modelSignals[i], this, proxySlots[i]);
+            disconnect(sourceModel(), modelSignals[i], this, proxySlots[i]);
     }
 
     QAbstractProxyModel::setSourceModel(_sourceModel);
diff --git a/src/kselectionproxymodel.cpp b/src/kselectionproxymodel.cpp
index 092f867..3c61a67 100644
--- a/src/kselectionproxymodel.cpp
+++ b/src/kselectionproxymodel.cpp
@@ -2215,9 +2215,9 @@ void KSelectionProxyModel::setSourceModel(QAbstractItemModel *_sourceModel)
       SLOT(sourceModelDestroyed())
     };
 
-    if (_sourceModel) {
+    if (sourceModel()) {
         for (int i = 0; i < int(sizeof modelSignals / sizeof *modelSignals); ++i)
-            disconnect(_sourceModel, modelSignals[i], this, proxySlots[i]);
+            disconnect(sourceModel(), modelSignals[i], this, proxySlots[i]);
     }
 
     // Must be called before QAbstractProxyModel::setSourceModel because it emit some signals.