diff --git a/erpnext/controllers/sales_and_purchase_return.py b/erpnext/controllers/sales_and_purchase_return.py
index e9633558ef..6054b9015a 100644
--- a/erpnext/controllers/sales_and_purchase_return.py
+++ b/erpnext/controllers/sales_and_purchase_return.py
@@ -24,10 +24,7 @@ def validate_return_against(doc):
 	else:
 		ref_doc = frappe.get_doc(doc.doctype, doc.return_against)
 
-		if doc.doctype in ("Sales Invoice", "Delivery Note"):
-			party_type = "customer"
-		else:
-			party_type = "supplier"
+		party_type == "customer" if doc.doctype in ("Sales Invoice", "Delivery Note") else "supplier"
 
 		if ref_doc.company == doc.company and ref_doc.get(pary_type) == doc.get(party_type) and ref_doc.docstatus == 1:
 			# validate posting date time