diff --git a/akka-stream-tests-tck/src/test/scala/akka/stream/tck/FusableProcessorTest.scala b/akka-stream-tests-tck/src/test/scala/akka/stream/tck/FusableProcessorTest.scala
index 4aba731323..9ea90c3f7f 100644
--- a/akka-stream-tests-tck/src/test/scala/akka/stream/tck/FusableProcessorTest.scala
+++ b/akka-stream-tests-tck/src/test/scala/akka/stream/tck/FusableProcessorTest.scala
@@ -3,9 +3,7 @@
  */
 package akka.stream.tck
 
-import akka.NotUsed
 import akka.stream._
-import akka.stream.impl.fusing.GraphStages
 import akka.stream.scaladsl.Flow
 import org.reactivestreams.Processor
 
@@ -17,8 +15,7 @@ class FusableProcessorTest extends AkkaIdentityProcessorVerification[Int] {
 
     implicit val materializer = ActorMaterializer(settings)(system)
 
-    // withAttributes "wraps" the underlying identity and protects it from automatic removal
-    Flow[Int].via(GraphStages.identity.asInstanceOf[Graph[FlowShape[Int, Int], NotUsed]]).named("identity").toProcessor.run()
+    Flow[Int].map(identity).toProcessor.run()
   }
 
   override def createElement(element: Int): Int = element