diff --git a/flink-streaming-java/src/test/java/org/apache/flink/streaming/api/graph/StreamGraphGeneratorTest.java b/flink-streaming-java/src/test/java/org/apache/flink/streaming/api/graph/StreamGraphGeneratorTest.java
index ec8eccae1eef5..dc01b6043e8fa 100644
--- a/flink-streaming-java/src/test/java/org/apache/flink/streaming/api/graph/StreamGraphGeneratorTest.java
+++ b/flink-streaming-java/src/test/java/org/apache/flink/streaming/api/graph/StreamGraphGeneratorTest.java
@@ -219,10 +219,10 @@ public void testOutputTypeConfigurationWithUdfStreamOperator() throws Exception
 
         DataStream<Integer> source = env.fromElements(1, 10);
 
-        NoOpUdfOperator<Integer> udfOperator = new NoOpUdfOperator(function);
+        NoOpUdfOperator<Integer> udfOperator = new NoOpUdfOperator<>(function);
 
         source.transform("no-op udf operator", BasicTypeInfo.INT_TYPE_INFO, udfOperator)
-                .addSink(new DiscardingSink());
+                .addSink(new DiscardingSink<>());
 
         env.getStreamGraph();