diff --git a/app/Main.hs b/app/Main.hs
index 8889ca4..683a8de 100644
--- a/app/Main.hs
+++ b/app/Main.hs
@@ -1,6 +1,4 @@
 module Main where
 
-import Rextra.Rextra (rextra)
-
 main :: IO ()
-main = rextra
+main = putStrLn "Nothing to see here"
diff --git a/src/Rextra/Automaton.hs b/src/Rextra/Automaton.hs
index dc8a334..49e07cd 100644
--- a/src/Rextra/Automaton.hs
+++ b/src/Rextra/Automaton.hs
@@ -153,7 +153,7 @@ initialPartition a =
 minimizeDfa :: (Ord s, Ord t) => Dfa.Dfa s t -> Dfa.Dfa (EquivalenceGroup s) t
 minimizeDfa a =
   let grouping      = findGroupingFixpoint (stateMap a) (initialPartition a)
-      mapping       = partitionToMap $ groupingToPartition grouping
+      mapping       = partitionToMap $ Map.keysSet grouping
       newEntryState = mapping Map.! entryState a
       newExitStates = Set.map (mapping Map.!) (exitStates a)
   in  fromJust $ fa grouping newEntryState newExitStates
diff --git a/src/Rextra/Rextra.hs b/src/Rextra/Rextra.hs
deleted file mode 100644
index 485f60b..0000000
--- a/src/Rextra/Rextra.hs
+++ /dev/null
@@ -1,6 +0,0 @@
--- This file will be split up as it grows.
-module Rextra.Rextra where
-
--- * Just a placeholder function
-rextra :: IO ()
-rextra = putStrLn "rextra"