1 files changed, 6 insertions, 2 deletions
diff --git a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala
index b988ba3..07557bc 100644
--- a/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala
+++ b/src/main/scala/com/tylerstonge/honeypot/ftp/FtpListener.scala
@@ -8,7 +8,11 @@ import akka.io.Tcp._
 import akka.io.{IO, Tcp}
 import akka.util.ByteString
 
-class FtpListener (port: Int)  extends Actor {
+object FtpListener {
+  def props(port: Int): Props = Props(new FtpListener(port))
+}
+
+class FtpListener(port: Int) extends Actor {
 
   val log: LoggingAdapter = Logging(context.system, this)
   IO(Tcp)(context.system) ! Bind(self, new InetSocketAddress("localhost", port))
@@ -17,7 +21,7 @@ class FtpListener (port: Int)  extends Actor {
     case Bound(localAddress) =>
       log.info("listening on {}", localAddress)
     case CommandFailed(_: Bind) => context.stop(self)
-    case Connected =>
+    case Connected(_, _) =>
       val handler = context.actorOf(Props[FtpHandler])
       val connection = sender()
       connection ! Register(handler)