diff --git a/tests/SMPAgentClient.hs b/tests/SMPAgentClient.hs
index ee67ef8f9..55e070230 100644
--- a/tests/SMPAgentClient.hs
+++ b/tests/SMPAgentClient.hs
@@ -123,7 +123,7 @@ cfg =
 withSmpAgentThreadOn :: (MonadUnliftIO m, MonadRandom m) => (ServiceName, String) -> (ThreadId -> m a) -> m a
 withSmpAgentThreadOn (port', db') =
   E.bracket
-    (forkIOWithUnmask ($ runSMPAgent cfg {tcpPort = port', dbFile = db'}))
+    (forkIO $ runSMPAgent cfg {tcpPort = port', dbFile = db'})
     (liftIO . killThread >=> const (removeFile db'))
 
 withSmpAgentOn :: (MonadUnliftIO m, MonadRandom m) => (ServiceName, String) -> m a -> m a
diff --git a/tests/SMPClient.hs b/tests/SMPClient.hs
index 6ca87db6d..cf30497ed 100644
--- a/tests/SMPClient.hs
+++ b/tests/SMPClient.hs
@@ -43,7 +43,7 @@ cfg =
 withSmpServerThreadOn :: (MonadUnliftIO m, MonadRandom m) => ServiceName -> (ThreadId -> m a) -> m a
 withSmpServerThreadOn port =
   E.bracket
-    (forkIOWithUnmask ($ runSMPServer cfg {tcpPort = port}))
+    (forkIO $ runSMPServer cfg {tcpPort = port})
     (liftIO . killThread)
 
 withSmpServerOn :: (MonadUnliftIO m, MonadRandom m) => ServiceName -> m a -> m a