diff --git a/tests/SMPProxyTests.hs b/tests/SMPProxyTests.hs
index 55c7f6929..35df65777 100644
--- a/tests/SMPProxyTests.hs
+++ b/tests/SMPProxyTests.hs
@@ -557,7 +557,7 @@ testProxyReconnectAfterRelayRestart =
       race_ (threadDelay 1000000) requestRelaySession
     requireProxyReconnect
 
-testProxyForwardTimeoutStuckSession :: IO ()
+testProxyForwardTimeoutStuckSession :: HasTestEnv => IO ()
 testProxyForwardTimeoutStuckSession =
   withSmpServerConfigOn (transport @TLS) proxyCfgForwardTimeout testPort $ \_ -> do
     g <- C.newRandom
@@ -591,7 +591,7 @@ testProxyRelayConcurrencyLimit =
       v : _ -> readMaybe v
       [] -> Nothing
 
-testForwardOversizedNotKept :: IO ()
+testForwardOversizedNotKept :: HasTestEnv => IO ()
 testForwardOversizedNotKept =
   withSmpServerConfigOn (transport @TLS) proxyCfg testPort $ \_ -> do
     g <- C.newRandom