Fix module dependency for older base package installs

HeinrichApfelmus · Dec 21, 2012 · 1aeaac0 · 1aeaac0
1 parent d7ead02
commit 1aeaac0
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/Data/Vault/ST_GHC.hs b/src/Data/Vault/ST_GHC.hs
@@ -9,8 +9,11 @@ import Data.IntMap (IntMap)
 import qualified Data.IntMap as IntMap
 import Data.IORef
 import Control.Monad.ST
-import Control.Monad.ST.Unsafe
-
+#if MIN_VERSION_base(4,4,0)
+import Control.Monad.ST.Unsafe as STUnsafe
+#else
+import Control.Monad.ST as STUnsafe
+#endif
 
 import Data.Unique.Really
 
@@ -38,7 +41,7 @@ empty :: Vault s
 empty = Vault Map.empty
 
 newKey :: ST s (Key s a)
-newKey = Control.Monad.ST.Unsafe.unsafeIOToST $ Key <$> newUnique
+newKey = STUnsafe.unsafeIOToST $ Key <$> newUnique
 
 lookup :: Key s a -> Vault s -> Maybe a
 lookup (Key k) (Vault m) = fromAny <$> Map.lookup k m