diff --git a/src/library/TestcontainersPostgresql.hs b/src/library/TestcontainersPostgresql.hs
--- a/src/library/TestcontainersPostgresql.hs
+++ b/src/library/TestcontainersPostgresql.hs
@@ -1,8 +1,8 @@
 module TestcontainersPostgresql
   ( Configs.Config.Config (..),
-    Configs.Distro.Distro (..),
     Configs.Auth.Auth (..),
     run,
+    setup,
   )
 where
 
@@ -12,7 +12,6 @@
 import qualified TestContainers.Hspec
 import qualified TestcontainersPostgresql.Configs.Auth as Configs.Auth
 import qualified TestcontainersPostgresql.Configs.Config as Configs.Config
-import qualified TestcontainersPostgresql.Configs.Distro as Configs.Distro
 import Prelude
 
 setup :: (TestContainers.MonadDocker m) => Configs.Config.Config -> m (Text, Int)
diff --git a/src/library/TestcontainersPostgresql/Configs/Config.hs b/src/library/TestcontainersPostgresql/Configs/Config.hs
--- a/src/library/TestcontainersPostgresql/Configs/Config.hs
+++ b/src/library/TestcontainersPostgresql/Configs/Config.hs
@@ -1,21 +1,22 @@
 module TestcontainersPostgresql.Configs.Config where
 
 import Data.Function
+import Data.Text (Text)
 import qualified Data.Text.Lazy as Text.Lazy
 import qualified TestContainers
 import qualified TestcontainersPostgresql.Configs.Auth as Configs.Auth
-import qualified TestcontainersPostgresql.Configs.Distro as Configs.Distro
 import Prelude
 
 data Config = Config
-  { forwardLogs :: Bool,
-    distro :: Configs.Distro.Distro,
-    auth :: Configs.Auth.Auth
+  { tagName :: Text,
+    auth :: Configs.Auth.Auth,
+    forwardLogs :: Bool
   }
 
 toContainerRequest :: Config -> TestContainers.ContainerRequest
-toContainerRequest (Config forwardLogs distro auth) =
-  Configs.Distro.toContainerRequest distro
+toContainerRequest (Config tagName auth forwardLogs) =
+  TestContainers.fromTag tagName
+    & TestContainers.containerRequest
     & TestContainers.setExpose [5432]
     & TestContainers.setWaitingFor waitUntilReady
     & Configs.Auth.updateContainerRequest auth
diff --git a/src/library/TestcontainersPostgresql/Configs/Distro.hs b/src/library/TestcontainersPostgresql/Configs/Distro.hs
deleted file mode 100644
--- a/src/library/TestcontainersPostgresql/Configs/Distro.hs
+++ /dev/null
@@ -1,40 +0,0 @@
-module TestcontainersPostgresql.Configs.Distro where
-
-import Data.Function
-import Data.Text (Text)
-import qualified TestContainers
-import Prelude
-
-data Distro
-  = Distro8
-  | Distro9
-  | Distro10
-  | Distro11
-  | Distro12
-  | Distro13
-  | Distro14
-  | Distro15
-  | Distro16
-  | Distro17
-  | Distro18
-  deriving stock (Show, Eq, Enum, Bounded, Ord)
-
-toTag :: Distro -> Text
-toTag = \case
-  Distro8 -> "postgres:8"
-  Distro9 -> "postgres:9"
-  Distro10 -> "postgres:10"
-  Distro11 -> "postgres:11"
-  Distro12 -> "postgres:12"
-  Distro13 -> "postgres:13"
-  Distro14 -> "postgres:14"
-  Distro15 -> "postgres:15"
-  Distro16 -> "postgres:16"
-  Distro17 -> "postgres:17"
-  Distro18 -> "postgres:18"
-
-toToImage :: Distro -> TestContainers.ToImage
-toToImage = TestContainers.fromTag . toTag
-
-toContainerRequest :: Distro -> TestContainers.ContainerRequest
-toContainerRequest = TestContainers.containerRequest . toToImage
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.0.4
+version: 0.1
 category: PostgreSQL, Codecs
 synopsis: Testcontainers integration for PostgreSQL
 homepage: https://github.com/nikita-volkov/testcontainers-postgresql
@@ -91,7 +91,6 @@
   other-modules:
     TestcontainersPostgresql.Configs.Auth
     TestcontainersPostgresql.Configs.Config
-    TestcontainersPostgresql.Configs.Distro
 
   build-depends:
     base >=4.11 && <5,