1 files changed, 6 insertions, 3 deletions
diff --git a/tests/Test/StarWars/Schema.hs b/tests/Test/StarWars/Schema.hs
index 0abda7b..daa000e 100644
--- a/tests/Test/StarWars/Schema.hs
+++ b/tests/Test/StarWars/Schema.hs
@@ -24,15 +24,18 @@ schema = hero :| [human, droid]
 hero :: MonadPlus m => Resolver m
 hero = Schema.objectA "hero" $ \case
   [] -> character artoo
-  [Argument "episode" (ValueInt n)]   -> character . getHero $ fromIntegral n
   [Argument "episode" (ValueEnum "NEWHOPE")] -> character $ getHero 4
   [Argument "episode" (ValueEnum "EMPIRE" )] -> character $ getHero 5
   [Argument "episode" (ValueEnum "JEDI"   )] -> character $ getHero 6
   _ -> ActionT $ throwE "Invalid arguments."
 
 human :: MonadPlus m => Resolver m
-human = Schema.objectA "human" $ \case
-  [Argument "id" (ValueString i)] -> character =<< lift (getHuman i)
+human = Schema.nullableObject "human" $ \case
+  [Argument "id" (ValueString i)] -> do
+      humanCharacter <- lift $ return $ getHuman i >>= Just
+      case humanCharacter of
+        Nothing -> return Nothing
+        Just e -> Just <$> character e
   _ -> ActionT $ throwE "Invalid arguments."
 
 droid :: MonadPlus m => Resolver m