Hack the test to type-check again

1 files changed, 3 insertions, 4 deletions

diff --git a/test/Main.hs b/test/Main.hs
index 4fdf4ad..209b88e 100644
--- a/test/Main.hs
+++ b/test/Main.hs
@@ -9,7 +9,6 @@ module Main where
 
 import Data.Dependent.EnumMap.Strict qualified as DE
 import Data.Dependent.Sum
-import Data.Some
 
 
 data Tag = A | B | C
@@ -22,9 +21,9 @@ data STag tag where
 deriving instance Show (STag tag)
 
 instance DE.Enum1 STag where
-  type Enum1Info STag = ()
-  fromEnum1 = \case { SA -> (0, ()); SB -> (1, ()); SC -> (2, ()) }
-  toEnum1 n () = case n of { 0 -> Some SA; 1 -> Some SB; 2 -> Some SC; _ -> error "invalid tag" }
+  type Enum1Info STag = STag
+  fromEnum1 = \case { SA -> (0, SA); SB -> (1, SB); SC -> (2, SC) }
+  toEnum1 n t = case (n, t) of { (0, SA) -> SA; (1, SB) -> SB; (2, SC) -> SC; _ -> error "invalid tag" }
 
 data Value tag where
   VA :: Int -> Value A