diff --git a/LICENSE b/LICENSE
--- a/LICENSE
+++ b/LICENSE
@@ -1,4 +1,4 @@
-Copyright (c) 2011-2017 Simon Hengel <sol@typeful.net>
+Copyright (c) 2011-2018 Simon Hengel <sol@typeful.net>
 Copyright (c) 2011-2012 Trystan Spangler <trystan.s@comcast.net>
 Copyright (c) 2011-2011 Greg Weber <greg@gregweber.info>
 
diff --git a/hspec-contrib.cabal b/hspec-contrib.cabal
--- a/hspec-contrib.cabal
+++ b/hspec-contrib.cabal
@@ -1,8 +1,8 @@
 name:             hspec-contrib
-version:          0.5.0
+version:          0.5.1
 license:          MIT
 license-file:     LICENSE
-copyright:        (c) 2011-2017 Simon Hengel,
+copyright:        (c) 2011-2018 Simon Hengel,
                   (c) 2014 Junji Hashimoto
 
 maintainer:       Simon Hengel <sol@typeful.net>
diff --git a/test/Test/Hspec/Contrib/HUnitSpec.hs b/test/Test/Hspec/Contrib/HUnitSpec.hs
--- a/test/Test/Hspec/Contrib/HUnitSpec.hs
+++ b/test/Test/Hspec/Contrib/HUnitSpec.hs
@@ -1,5 +1,5 @@
 {-# OPTIONS_GHC -fno-warn-orphans #-}
-module Test.Hspec.Contrib.HUnitSpec (main, spec) where
+module Test.Hspec.Contrib.HUnitSpec (spec) where
 
 import           Helper
 
@@ -7,17 +7,8 @@
 import           Test.Hspec.Contrib.HUnit
 import           Test.HUnit
 
-main :: IO ()
-main = hspec spec
-
-shouldYield :: Test -> [Tree (ActionWith ()) String] -> Expectation
-a `shouldYield` b = map (Blind . fmap itemRequirement) <$> runSpecM (fromHUnitTest a) `shouldReturn` map Blind b
-
-instance Eq a => Eq (Tree c a) where
-  x == y = case (x, y) of
-    (Node s1 xs, Node s2 ys) -> s1 == s2 && xs == ys
-    (Leaf x1, Leaf x2) -> x1 == x2
-    _ -> False
+shouldYield :: Test -> [Tree () String] -> Expectation
+a `shouldYield` b = map (bimapTree (const ()) itemRequirement) <$> runSpecM (fromHUnitTest a) `shouldReturn` b
 
 spec :: Spec
 spec = do
diff --git a/test/Test/Hspec/Contrib/RetrySpec.hs b/test/Test/Hspec/Contrib/RetrySpec.hs
--- a/test/Test/Hspec/Contrib/RetrySpec.hs
+++ b/test/Test/Hspec/Contrib/RetrySpec.hs
@@ -1,11 +1,8 @@
-module Test.Hspec.Contrib.RetrySpec (main, spec) where
+module Test.Hspec.Contrib.RetrySpec (spec) where
 
 import           Helper
 
 import           Test.Hspec.Contrib.Retry
-
-main :: IO ()
-main = hspec spec
 
 spec :: Spec
 spec = do