1 files changed, 5 insertions, 1 deletions

diff --git a/tests/Data/Torrent/InfoHashSpec.hs b/tests/Data/Torrent/InfoHashSpec.hs
index ba9ce9a4..afa017ef 100644
--- a/tests/Data/Torrent/InfoHashSpec.hs
+++ b/tests/Data/Torrent/InfoHashSpec.hs
@@ -2,6 +2,8 @@
 module Data.Torrent.InfoHashSpec (spec) where
 
 import Control.Applicative
+import Data.ByteString as BS
+import Data.Maybe
 import System.FilePath
 import Test.Hspec
 import Test.QuickCheck
@@ -12,7 +14,9 @@ import Data.Torrent.InfoHash as IH
 
 
 instance Arbitrary InfoHash where
-  arbitrary = IH.hash <$> arbitrary
+  arbitrary = do
+    bs <- BS.pack <$> vectorOf 20 arbitrary
+    pure $ fromMaybe (error "arbitrary infohash") $ byteStringToInfoHash bs
 
 type TestPair = (FilePath, String)