1 files changed, 2 insertions, 3 deletions

diff --git a/src/Data/JLD.hs b/src/Data/JLD.hs
index a7042dc..40abae1 100644
--- a/src/Data/JLD.hs
+++ b/src/Data/JLD.hs
@@ -26,8 +26,7 @@ import Data.JLD.Options
 import Data.JLD.Util (flattenSingletonArray, valueToArray)
 
 import Data.Aeson (Value (..))
-import Data.Aeson.KeyMap qualified as KM
-import Data.Vector qualified as V (singleton)
+import Data.Aeson.KeyMap qualified as KM (lookup, size)
 import Text.URI (URI)
 
 data JLDExpansionParams e m = JLDExpansionParams
@@ -72,7 +71,7 @@ expand document baseUrl paramsFn = do
             jldExpansionParamsExpandContext <&> flattenSingletonArray .> \case
                 Array expandedContext -> Array expandedContext
                 (Object expandedContext) | Just ctx <- KM.lookup (show KeywordContext) expandedContext -> ctx
-                expandedContext -> Array <| V.singleton expandedContext
+                expandedContext -> Array <| pure expandedContext
 
     activeContext' <- case maybeExpandContext of
         Just expandContext ->