diff --git a/src/Graphula/Key.hs b/src/Graphula/Key.hs
index 60c0a6e..deec63a 100644
--- a/src/Graphula/Key.hs
+++ b/src/Graphula/Key.hs
@@ -6,9 +6,8 @@
 
 -- | Convenience functions for working with 'Key' dependencies
 module Graphula.Key
-  ( onlyKey
-  , keys
-  , Keys
+  ( EntityKeys(..)
+  , onlyKey
   ) where
 
 import Database.Persist