X-Git-Url: http://git.megacz.com/?p=ghc-hetmet.git;a=blobdiff_plain;f=compiler%2Fmain%2FHscTypes.lhs;h=e79acf4775c380b84f0e84198ed48bdd0c5535a6;hp=c187932835cec9a8842d388b63587bc6e11a6baa;hb=f74cf811fae9a09799b56a24f53cb55b748b57f8;hpb=1dfe7f1ee44b6ae59e072eec720984b136fc1dc5

diff --git a/compiler/main/HscTypes.lhs b/compiler/main/HscTypes.lhs
index c187932..e79acf4 100644
--- a/compiler/main/HscTypes.lhs
+++ b/compiler/main/HscTypes.lhs
@@ -480,7 +480,8 @@ withLocalCallbacks f m = do
   let cb' = f cb0
   setSession (hsc_env { hsc_callbacks = cb' `seq` cb' })
   r <- m
-  setSession (hsc_env { hsc_callbacks = cb0 })
+  hsc_env' <- getSession
+  setSession (hsc_env' { hsc_callbacks = cb0 })
   return r
 
 \end{code}