diff --git a/documentation/src/test/java/example/extensions/RandomNumberExtension.java b/documentation/src/test/java/example/extensions/RandomNumberExtension.java
index 3bbece08e206..b2a900dc5ea5 100644
--- a/documentation/src/test/java/example/extensions/RandomNumberExtension.java
+++ b/documentation/src/test/java/example/extensions/RandomNumberExtension.java
@@ -86,7 +86,7 @@ private void injectFields(Class<?> testClass, Object testInstance,
 	}
 
 	private static boolean isInteger(Class<?> type) {
-		return int.class.isAssignableFrom(type);
+		return type == Integer.class || type == int.class;
 	}
 
 }