diff --git a/classes/Aircraft.php b/classes/Aircraft.php
index 6e32f64..8ea3d98 100644
--- a/classes/Aircraft.php
+++ b/classes/Aircraft.php
@@ -68,12 +68,12 @@ class Aircraft extends GraphQLSubscription
    * @param  \GraphQL\Type\Definition\ResolveInfo  $resolveInfo
    * @return mixed
    */
-  public function resolve($root, array $args, GraphQLContext $context, ResolveInfo $resolveInfo): array
+  public function resolve($root, array $args, GraphQLContext $context, ResolveInfo $resolveInfo): Model
   {
     // Optionally manipulate the `$root` item before it gets broadcasted to
     // subscribed client(s).
-    //$root->load();
+    $root->load();
 
-    return array($root);
+    return $root;
   }
 }