diff --git a/pontarius-xmpp.cabal b/pontarius-xmpp.cabal
index f8d2c3a..52294ec 100644
--- a/pontarius-xmpp.cabal
+++ b/pontarius-xmpp.cabal
@@ -154,6 +154,8 @@ Test-Suite tests
   Other-modules: Tests.Arbitrary
                , Tests.Arbitrary.Xml
                , Tests.Arbitrary.Xmpp
+               , Tests.Parsers
+               , Tests.Picklers
   ghc-options: -Wall -O2 -fno-warn-orphans
 
 Test-Suite doctest
diff --git a/tests/Run/SendReceive.hs b/tests/Run/SendReceive.hs
index 8f4794b..1b326d4 100644
--- a/tests/Run/SendReceive.hs
+++ b/tests/Run/SendReceive.hs
@@ -60,7 +60,7 @@ run = void $ do
             Nothing -> UseRealm
             Just srv -> case port of
                 Nothing -> UseSrv srv
-                Just p -> UseHost srv (PortNumber $ fromIntegral p)
+                Just p -> UseHost srv (fromIntegral p)
     _ <- configuredLoglevel conf
     mbSess1 <- session realm (simpleAuth uname1 pwd1)
                                 ((xmppConfig conDetails))