diff --git a/core/src/main/kotlin/net/corda/core/node/services/Services.kt b/core/src/main/kotlin/net/corda/core/node/services/Services.kt
index 800040174c..8d90ba7c77 100644
--- a/core/src/main/kotlin/net/corda/core/node/services/Services.kt
+++ b/core/src/main/kotlin/net/corda/core/node/services/Services.kt
@@ -203,8 +203,7 @@ interface VaultService {
 
 inline fun <reified T : LinearState> VaultService.linearHeadsOfType() = linearHeadsOfType_(T::class.java)
 inline fun <reified T : DealState> VaultService.dealsWith(party: Party) = linearHeadsOfType<T>().values.filter {
-    // TODO: Replace name comparison with full party comparison (keys are currenty not equal)
-    it.state.data.parties.any { it.name == party.name }
+    it.state.data.parties.any { it == party }
 }
 
 /**