diff --git a/lib/CodeGen/RegAllocLinearScan.cpp b/lib/CodeGen/RegAllocLinearScan.cpp
index 202a152..d04bb69 100644
--- a/lib/CodeGen/RegAllocLinearScan.cpp
+++ b/lib/CodeGen/RegAllocLinearScan.cpp
@@ -340,13 +340,11 @@
         unsigned rep = li_->rep(reg);
 
         assert((MRegisterInfo::isPhysicalRegister(rep) ||
-                v2pMap_.find(rep) != v2pMap_.end() ||
-                v2ssMap_.find(rep) != v2ssMap_.end()) &&
+                v2pMap_.count(rep) || v2ssMap_.count(rep)) &&
                "representative register is not allocated!");
 
         assert(MRegisterInfo::isVirtualRegister(reg) &&
-               v2pMap_.find(reg) == v2pMap_.end() &&
-               v2ssMap_.find(reg) == v2ssMap_.end() &&
+               !v2pMap_.count(reg) && !v2ssMap_.count(reg) &&
                "coalesced register is already allocated!");
 
         if (MRegisterInfo::isPhysicalRegister(rep)) {
@@ -772,7 +770,7 @@
            "attempt to assign stack slot to already assigned register?");
     // if the virtual register was previously assigned clear the mapping
     // and free the virtual register
-    if (v2pMap_.find(virtReg) != v2pMap_.end()) {
+    if (v2pMap_.count(virtReg)) {
         clearVirtReg(virtReg);
     }
 }