diff --git a/akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala b/akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala
index e983977635..60c36cf470 100644
--- a/akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala
+++ b/akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala
@@ -90,8 +90,11 @@ class GraphStageDocSpec extends AkkaSpec {
 
   "Demonstrate creation of GraphStage Sink" in {
     //#custom-sink-example
+    import akka.stream.Attributes
+    import akka.stream.Inlet
     import akka.stream.SinkShape
     import akka.stream.stage.GraphStage
+    import akka.stream.stage.GraphStageLogic
     import akka.stream.stage.InHandler
 
     class StdoutSink extends GraphStage[SinkShape[Int]] {