diff --git a/build.sbt b/build.sbt
index c627031b9..a69615490 100644
--- a/build.sbt
+++ b/build.sbt
@@ -26,7 +26,7 @@ lazy val versions = new {
   val circeOptics = "0.14.1"
   val diffx = "0.7.0"
   val dockerClient = "8.16.0"
-  val dockerTestkit = "0.9.9"
+  val dockerTestkit = "0.11.0-beta1"
   val doobie = "1.0.0-RC2"
   val enumeratum = "1.7.0"
   val enumeratumDoobie = "1.7.1"
diff --git a/src/test/scala/io/iohk/atala/prism/node/repositories/DockerPostgresService.scala b/src/test/scala/io/iohk/atala/prism/node/repositories/DockerPostgresService.scala
index 955c785e5..56c6dd9e8 100644
--- a/src/test/scala/io/iohk/atala/prism/node/repositories/DockerPostgresService.scala
+++ b/src/test/scala/io/iohk/atala/prism/node/repositories/DockerPostgresService.scala
@@ -79,11 +79,12 @@ object DockerPostgresService extends DockerKit {
         container: DockerContainerState
     )(implicit
         dockerExecutor: DockerCommandExecutor,
-        ec: ExecutionContext
+        ec: ExecutionContext,
+        timeout: Duration
     ): Future[Boolean] = {
 
       container
-        .getPorts()(dockerExecutor, ec)
+        .getPorts()(dockerExecutor, ec, timeout)
         .map { _ =>
           try {
             Class.forName("org.postgresql.Driver")