diff --git a/accelerate-io-array/src/Data/Array/Accelerate/IO/Data/Primitive/ByteArray.hs b/accelerate-io-array/src/Data/Array/Accelerate/IO/Data/Primitive/ByteArray.hs
index 73d2734..4df4b1b 100644
--- a/accelerate-io-array/src/Data/Array/Accelerate/IO/Data/Primitive/ByteArray.hs
+++ b/accelerate-io-array/src/Data/Array/Accelerate/IO/Data/Primitive/ByteArray.hs
@@ -18,6 +18,7 @@ module Data.Array.Accelerate.IO.Data.Primitive.ByteArray
 import Data.Primitive.ByteArray
 
 import GHC.Base
+import GHC.Exts
 import GHC.ForeignPtr
 
 
diff --git a/accelerate-io-vector/src/Data/Array/Accelerate/IO/Data/Primitive/ByteArray.hs b/accelerate-io-vector/src/Data/Array/Accelerate/IO/Data/Primitive/ByteArray.hs
index 73d2734..4df4b1b 100644
--- a/accelerate-io-vector/src/Data/Array/Accelerate/IO/Data/Primitive/ByteArray.hs
+++ b/accelerate-io-vector/src/Data/Array/Accelerate/IO/Data/Primitive/ByteArray.hs
@@ -18,6 +18,7 @@ module Data.Array.Accelerate.IO.Data.Primitive.ByteArray
 import Data.Primitive.ByteArray
 
 import GHC.Base
+import GHC.Exts
 import GHC.ForeignPtr