diff --git a/om-fork.cabal b/om-fork.cabal
--- a/om-fork.cabal
+++ b/om-fork.cabal
@@ -1,6 +1,6 @@
 cabal-version:       3.0
 name:                om-fork
-version:             0.7.1.6
+version:             0.7.1.7
 synopsis:            Concurrency utilities.
 description:         Actor pattern and some limited structured concurrency
                      tools
@@ -21,7 +21,7 @@
     , aeson        >= 2.0.3.0  && < 2.1
     , base         >= 4.15.0.0 && < 4.16
     , exceptions   >= 0.10.4   && < 0.11
-    , ki           >= 0.2.0.1  && < 0.3
+    , ki           >= 1.0.0    && < 1.1
     , monad-logger >= 0.3.36   && < 0.4
     , om-show      >= 0.1.2.0  && < 0.2
     , text         >= 1.2.5.0  && < 1.3
diff --git a/src/OM/Fork.hs b/src/OM/Fork.hs
--- a/src/OM/Fork.hs
+++ b/src/OM/Fork.hs
@@ -38,6 +38,7 @@
 import Data.Aeson (ToJSON, toJSON)
 import Data.String (IsString)
 import Data.Text (Text)
+import GHC.Conc (atomically)
 import OM.Show (showt)
 import UnliftIO (MonadUnliftIO, askRunInIO, throwString)
 import qualified Ki
@@ -208,6 +209,6 @@
 
 {- | Wait for all racing threads to terminate. -}
 wait :: (MonadIO m, Race) => m ()
-wait = liftIO $ Ki.wait ?scope
+wait = liftIO . atomically $ Ki.awaitAll ?scope