diff --git a/src/main/java/com/reandroid/dex/model/DexClassRepository.java b/src/main/java/com/reandroid/dex/model/DexClassRepository.java
index 5fb7719b8..58eb1e63f 100644
--- a/src/main/java/com/reandroid/dex/model/DexClassRepository.java
+++ b/src/main/java/com/reandroid/dex/model/DexClassRepository.java
@@ -113,6 +113,12 @@ default boolean contains(Key key){
         if(key instanceof TypeListKey){
             return contains(SectionType.TYPE_LIST, key);
         }
+        if(key instanceof MethodHandleKey){
+            return contains(SectionType.METHOD_HANDLE, key);
+        }
+        if(key instanceof CallSiteKey){
+            return contains(SectionType.CALL_SITE_ID, key);
+        }
         throw new IllegalArgumentException("Unknown key type: " + key.getClass() + ", '" + key + "'");
     }
     default boolean containsClass(TypeKey key){