1 files changed, 2 insertions, 2 deletions
diff --git a/agent/src/test/java/com/code_intelligence/jazzer/autofuzz/TestHelpers.java b/agent/src/test/java/com/code_intelligence/jazzer/autofuzz/TestHelpers.java
index 52f19a74..d556beb3 100644
--- a/agent/src/test/java/com/code_intelligence/jazzer/autofuzz/TestHelpers.java
+++ b/agent/src/test/java/com/code_intelligence/jazzer/autofuzz/TestHelpers.java
@@ -24,6 +24,7 @@ import java.io.ByteArrayInputStream;
 import java.lang.reflect.Constructor;
 import java.lang.reflect.Executable;
 import java.lang.reflect.Method;
+import java.lang.reflect.Type;
 import java.util.List;
 
 class TestHelpers {
@@ -57,8 +58,7 @@ class TestHelpers {
   }
 
   static void consumeTestCase(
-      Class<?> type, Object expectedResult, String expectedResultString, List<Object> cannedData) {
-    assertTrue(expectedResult == null || type.isAssignableFrom(expectedResult.getClass()));
+      Type type, Object expectedResult, String expectedResultString, List<Object> cannedData) {
     AutofuzzCodegenVisitor visitor = new AutofuzzCodegenVisitor();
     FuzzedDataProvider data = CannedFuzzedDataProvider.create(cannedData);
     assertGeneralEquals(expectedResult, Meta.consume(data, type, visitor));