From 7be4090ccd35464bd8ce0f5a5badab10649ec17f Mon Sep 17 00:00:00 2001
From: jczuchnowski <jakub.czuchnowski@gmail.com>
Date: Sun, 26 Aug 2018 00:50:32 +0200
Subject: [PATCH] Add missing imports for Custom stream processing docs #25525

---
 akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala b/akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala
index b5e247eac5..7be3e21a89 100644
--- a/akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala
+++ b/akka-docs/src/test/scala/docs/stream/GraphStageDocSpec.scala
@@ -40,9 +40,11 @@ class GraphStageDocSpec extends AkkaSpec {
 
   "Demonstrate creation of GraphStage Source" in {
     //#custom-source-example
+    import akka.stream.Attributes
+    import akka.stream.Outlet
     import akka.stream.SourceShape
-    import akka.stream.Graph
     import akka.stream.stage.GraphStage
+    import akka.stream.stage.GraphStageLogic
     import akka.stream.stage.OutHandler
 
     class NumbersSource extends GraphStage[SourceShape[Int]] {