diff --git a/yesod-core/Yesod/Core/Handler.hs b/yesod-core/Yesod/Core/Handler.hs
index 91ca3d8a..54b13650 100644
--- a/yesod-core/Yesod/Core/Handler.hs
+++ b/yesod-core/Yesod/Core/Handler.hs
@@ -205,7 +205,7 @@ import           Control.Monad.Trans.Resource  (MonadResource, InternalState, ru
 import qualified System.PosixCompat.Files as PC
 #endif
 #if MIN_VERSION_wai(2, 1, 0)
-import Control.Monad.Trans.Control (control)
+import Control.Monad.Trans.Control (control, MonadBaseControl)
 #endif
 import Data.Conduit (Source, transPipe, Flush (Flush), yield, Producer
 #if MIN_VERSION_wai(2, 1, 0)