1 files changed, 4 insertions, 2 deletions
diff --git a/Hsbot/Main.hs b/Hsbot/Main.hs
index ea1bf16..592fefa 100644
--- a/Hsbot/Main.hs
+++ b/Hsbot/Main.hs
@@ -2,7 +2,9 @@ module Hsbot.Main
     ( imain
     ) where
 
+import Control.Exception
 import Control.Monad.State
+import Prelude hiding (catch)
 import System.IO
 
 import Config
@@ -15,8 +17,8 @@ import Hsbot.Types
 imain :: IO ()
 imain = do
     bot <- connectServer $ ircServer config
-    bot' <- (execStateT run bot) `catch` (const $ return bot)
-    evalStateT disconnectServer bot'
+    (execStateT run bot) `catch` (\(ex :: IOException) -> return bot)
+    evalStateT disconnectServer bot
 
 -- | The Bot monad main function
 run :: IrcBot ()