diff --git a/src/library/TestcontainersPostgresql.hs b/src/library/TestcontainersPostgresql.hs
--- a/src/library/TestcontainersPostgresql.hs
+++ b/src/library/TestcontainersPostgresql.hs
@@ -8,18 +8,20 @@
 
 import Data.Function
 import Data.Text (Text)
+import Data.Word (Word16)
 import qualified TestContainers
 import qualified TestContainers.Hspec
 import qualified TestcontainersPostgresql.Configs.Auth as Configs.Auth
 import qualified TestcontainersPostgresql.Configs.Config as Configs.Config
 import Prelude
 
-setup :: (TestContainers.MonadDocker m) => Configs.Config.Config -> m (Text, Int)
+setup :: (TestContainers.MonadDocker m) => Configs.Config.Config -> m (Text, Word16)
 setup config = do
   container <- TestContainers.run (Configs.Config.toContainerRequest config)
-  pure $ TestContainers.containerAddress container 5432
+  let (host, port) = TestContainers.containerAddress container 5432
+  pure (host, fromIntegral port)
 
 -- | Run a session on a PostgreSQL container in the scope of a host and a port.
-run :: Configs.Config.Config -> ((Text, Int) -> IO ()) -> IO ()
+run :: Configs.Config.Config -> ((Text, Word16) -> IO ()) -> IO ()
 run config = do
   TestContainers.Hspec.withContainers (setup config)
diff --git a/testcontainers-postgresql.cabal b/testcontainers-postgresql.cabal
--- a/testcontainers-postgresql.cabal
+++ b/testcontainers-postgresql.cabal
@@ -1,6 +1,6 @@
 cabal-version: 3.0
 name: testcontainers-postgresql
-version: 0.1
+version: 0.2
 category: PostgreSQL, Codecs
 synopsis: Testcontainers integration for PostgreSQL
 homepage: https://github.com/nikita-volkov/testcontainers-postgresql
