diff --git a/KeyboardLayoutGuide/KeyboardLayoutGuide/Keyboard+LayoutGuide.swift b/KeyboardLayoutGuide/KeyboardLayoutGuide/Keyboard+LayoutGuide.swift
index beb2aaa..357f81c 100644
--- a/KeyboardLayoutGuide/KeyboardLayoutGuide/Keyboard+LayoutGuide.swift
+++ b/KeyboardLayoutGuide/KeyboardLayoutGuide/Keyboard+LayoutGuide.swift
@@ -98,7 +98,7 @@ open class KeyboardLayoutGuide: UILayoutGuide {
 extension UILayoutGuide {
     internal var heightConstraint: NSLayoutConstraint? {
         return owningView?.constraints.first {
-            $0 == self && $0.firstAttribute == .height
+            $0.firstItem as? UILayoutGuide == self && $0.firstAttribute == .height
         }
     }
 }