diff --git a/neat/src/test/scala/com/lunatech/neat4s/MapListElementsToOutletsSpec.scala b/neat/src/test/scala/com/lunatech/neat4s/akkastreams/MapListElementsToOutletsSpec.scala
similarity index 95%
rename from neat/src/test/scala/com/lunatech/neat4s/MapListElementsToOutletsSpec.scala
rename to neat/src/test/scala/com/lunatech/neat4s/akkastreams/MapListElementsToOutletsSpec.scala
index c589ab9..74f8ae6 100644
--- a/neat/src/test/scala/com/lunatech/neat4s/MapListElementsToOutletsSpec.scala
+++ b/neat/src/test/scala/com/lunatech/neat4s/akkastreams/MapListElementsToOutletsSpec.scala
@@ -1,9 +1,10 @@
-package com.lunatech.neat4s
+package com.lunatech.neat4s.akkastreams
 
 import akka.NotUsed
 import akka.actor.testkit.typed.scaladsl.ScalaTestWithActorTestKit
 import akka.stream.SourceShape
 import akka.stream.scaladsl.{GraphDSL, Merge, Sink, Source}
+import com.lunatech.neat4s.MapListElementsToOutputs
 import org.scalatest.concurrent.ScalaFutures
 import org.scalatest.matchers.should.Matchers
 import org.scalatest.wordspec.AnyWordSpecLike
diff --git a/neat/src/test/scala/com/lunatech/neat4s/NetworkNodeSpec.scala b/neat/src/test/scala/com/lunatech/neat4s/akkastreams/NetworkNodeSpec.scala
similarity index 97%
rename from neat/src/test/scala/com/lunatech/neat4s/NetworkNodeSpec.scala
rename to neat/src/test/scala/com/lunatech/neat4s/akkastreams/NetworkNodeSpec.scala
index b6e366b..c49c57d 100644
--- a/neat/src/test/scala/com/lunatech/neat4s/NetworkNodeSpec.scala
+++ b/neat/src/test/scala/com/lunatech/neat4s/akkastreams/NetworkNodeSpec.scala
@@ -1,10 +1,10 @@
-package com.lunatech.neat4s
+package com.lunatech.neat4s.akkastreams
 
 import akka.NotUsed
 import akka.actor.testkit.typed.scaladsl.ScalaTestWithActorTestKit
 import akka.stream.SourceShape
 import akka.stream.scaladsl.{GraphDSL, Merge, Sink, Source}
-import com.lunatech.neat4s.akkastreams.NetworkNode
+import com.lunatech.neat4s.ActivationFunction
 import org.scalatest.matchers.should.Matchers
 import org.scalatest.wordspec.AnyWordSpecLike