diff --git a/infrastructure/ssz/src/testFixtures/java/tech/pegasys/teku/infrastructure/ssz/RandomSszDataGenerator.java b/infrastructure/ssz/src/testFixtures/java/tech/pegasys/teku/infrastructure/ssz/RandomSszDataGenerator.java
index a5252163439..02d87211ba3 100644
--- a/infrastructure/ssz/src/testFixtures/java/tech/pegasys/teku/infrastructure/ssz/RandomSszDataGenerator.java
+++ b/infrastructure/ssz/src/testFixtures/java/tech/pegasys/teku/infrastructure/ssz/RandomSszDataGenerator.java
@@ -122,10 +122,9 @@ public <T extends SszData> Stream<T> randomDataStream(final SszSchema<T> schema)
             SszCollection<SszData> ret = collectionSchema.createFromElements(children);
             return (T) ret;
           });
-    } else if (schema instanceof SszUnionSchema) {
+    } else if (schema instanceof SszUnionSchema<?> unionSchema) {
       return Stream.generate(
           () -> {
-            SszUnionSchema<?> unionSchema = (SszUnionSchema<?>) schema;
             int selector = random.nextInt(unionSchema.getTypesCount());
             return (T)
                 unionSchema.createFromValue(