Domain -> ByteString.

1 files changed, 5 insertions, 3 deletions

diff --git a/Network/DNS/StateBinary.hs b/Network/DNS/StateBinary.hs
index 5099104..bd54a75 100644
--- a/Network/DNS/StateBinary.hs
+++ b/Network/DNS/StateBinary.hs
@@ -3,6 +3,8 @@ module Network.DNS.StateBinary where
 import Blaze.ByteString.Builder
 import Control.Monad.State
 import Data.Binary.Get
+import Data.ByteString (ByteString)
+import qualified Data.ByteString.Char8 as BS
 import qualified Data.ByteString.Lazy.Char8 as L
 import Data.Char
 import Data.Int
@@ -85,10 +87,10 @@ getPosition = fromIntegral <$> lift bytesRead
 getNBytes :: Int -> SGet [Int]
 getNBytes len = toInts <$> getNByteString len
   where
-    toInts = map ord . L.unpack
+    toInts = map ord . BS.unpack
 
-getNByteString :: Int -> SGet L.ByteString
-getNByteString len = lift . getLazyByteString . fromIntegral $ len
+getNByteString :: Int -> SGet ByteString
+getNByteString = lift . getByteString . fromIntegral
 
 ----------------------------------------------------------------