diff --git a/drivers/virt/haven/hh_dbl.c b/drivers/virt/haven/hh_dbl.c
index d08a27340482..04a108bcfac5 100644
--- a/drivers/virt/haven/hh_dbl.c
+++ b/drivers/virt/haven/hh_dbl.c
@@ -443,6 +443,9 @@ int hh_dbl_tx_unregister(void *dbl_client_desc)
 		return -EINVAL;
 	}
 
+	pr_debug("%s: Unregistering client for label: %d\n",
+			__func__, client_desc->label);
+
 	/* Rx client still holding the "client_desc". Do not remove now. */
 	if (!cap_table_entry->rx_reg_done) {
 		cap_table_entry->client_desc = NULL;
@@ -454,9 +457,6 @@ int hh_dbl_tx_unregister(void *dbl_client_desc)
 	cap_table_entry->tx_reg_done = 0;
 	mutex_unlock(&cap_table_entry->cap_entry_lock);
 
-	pr_debug("%s: Unregistered client for label: %d\n",
-			__func__, client_desc->label);
-
 	return 0;
 }
 EXPORT_SYMBOL(hh_dbl_tx_unregister);
@@ -494,6 +494,9 @@ int hh_dbl_rx_unregister(void *dbl_client_desc)
 		return -EINVAL;
 	}
 
+	pr_debug("%s: Unregistering client for label: %d\n", __func__,
+			 client_desc->label);
+
 	/* Tx client still holding the "client_desc". Do not remove now.*/
 	if (!cap_table_entry->tx_reg_done) {
 		cap_table_entry->client_desc = NULL;
@@ -508,9 +511,6 @@ int hh_dbl_rx_unregister(void *dbl_client_desc)
 
 	mutex_unlock(&cap_table_entry->cap_entry_lock);
 
-	pr_debug("%s: Unregistered client for label: %d\n", __func__,
-		 client_desc->label);
-
 	return 0;
 }
 EXPORT_SYMBOL(hh_dbl_rx_unregister);