diff --git a/Language/Haskell/GhcMod/Gap.hs b/Language/Haskell/GhcMod/Gap.hs
index 6f699f5..f97eafe 100644
--- a/Language/Haskell/GhcMod/Gap.hs
+++ b/Language/Haskell/GhcMod/Gap.hs
@@ -198,7 +198,7 @@ withContext :: Ghc a -> Ghc a
 withContext action = gbracket setup teardown body
   where
     setup = getContext
-    teardown = setContext
+    teardown = setCtx
     body _ = do
         topImports >>= setCtx
         action