diff --git a/CHANGELOG.md b/CHANGELOG.md
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+#### 0.2.0.1
+
+* Change `JSONSchema` instances to `Map`
+
 ## 0.2
 
 * Fix loop in ToJSON instances for StringHashMaps
diff --git a/rest-stringmap.cabal b/rest-stringmap.cabal
--- a/rest-stringmap.cabal
+++ b/rest-stringmap.cabal
@@ -1,5 +1,5 @@
 name:                rest-stringmap
-version:             0.2
+version:             0.2.0.1
 license:             BSD3
 synopsis:            Maps with stringy keys that can be transcoded to JSON and XML.
 description:         Maps with stringy keys that can be transcoded to JSON and XML.
@@ -30,7 +30,7 @@
     , containers == 0.5.*
     , hashable == 1.2.*
     , hxt == 9.3.*
-    , json-schema == 0.4.*
+    , json-schema == 0.5.*
     , tagged >= 0.2 && < 0.8
     , text >= 0.10 && < 1.2
     , tostring == 0.2.*
diff --git a/src/Rest/StringMap/Util.hs b/src/Rest/StringMap/Util.hs
--- a/src/Rest/StringMap/Util.hs
+++ b/src/Rest/StringMap/Util.hs
@@ -4,8 +4,7 @@
   , mapSchema
   ) where
 
-import Data.JSON.Schema (JSONSchema, Schema, schema)
-import Data.JSON.Schema.Combinators (field)
+import Data.JSON.Schema (JSONSchema, Schema, Value (Map), schema)
 import Data.Proxy (Proxy)
 import Data.String (IsString (..))
 import Data.String.ToString (ToString (..))
@@ -21,4 +20,4 @@
 pickleMap mapKeys mapKeys' = xpWrap (mapKeys fromString, mapKeys' toString) xpickle
 
 mapSchema :: JSONSchema a => Proxy a -> Schema
-mapSchema = field "key" False . schema
+mapSchema = Map . schema
