diff --git a/src/GraphQL/Types/BeamType.php b/src/GraphQL/Types/BeamType.php
index 9a697f8..1954a74 100644
--- a/src/GraphQL/Types/BeamType.php
+++ b/src/GraphQL/Types/BeamType.php
@@ -47,7 +47,6 @@ public function fields(): array
             'code' => [
                 'type' => GraphQL::type('String!'),
                 'description' => __('enjin-platform-beam::mutation.claim_beam.args.code'),
-                'excludeFrom' => ['GetBeam', 'GetBeams'],
             ],
             ...$this->getCommonFields(),
             'collection' => [
@@ -92,7 +91,6 @@ public function fields(): array
                 },
                 'selectable' => false,
                 'is_relation' => false,
-                'excludeFrom' => ['GetBeam', 'GetBeams'],
             ],
             'probabilities' => [
                 'type' => GraphQL::type('Object'),
diff --git a/tests/Feature/GraphQL/Queries/GetBeamTest.php b/tests/Feature/GraphQL/Queries/GetBeamTest.php
index 608388e..31a045f 100644
--- a/tests/Feature/GraphQL/Queries/GetBeamTest.php
+++ b/tests/Feature/GraphQL/Queries/GetBeamTest.php
@@ -179,7 +179,7 @@ public function test_it_hides_code_field_when_unauthenticated()
         ]);
 
         $response = $this->graphql($this->method, ['code' => $this->beam->code], true);
-        $this->assertEquals('Cannot query field "code" on type "Beam".', $response['error']);
+        $this->assertEquals('Cannot query field "code" on type "BeamClaim".', $response['error']);
 
         config([
             'enjin-platform.auth' => null,
diff --git a/tests/Feature/GraphQL/Queries/GetBeamsTest.php b/tests/Feature/GraphQL/Queries/GetBeamsTest.php
index 74a3264..50fe2c4 100644
--- a/tests/Feature/GraphQL/Queries/GetBeamsTest.php
+++ b/tests/Feature/GraphQL/Queries/GetBeamsTest.php
@@ -96,7 +96,7 @@ public function test_it_hides_code_field_when_unauthenticated()
         ]);
 
         $response = $this->graphql($this->method, [], true);
-        $this->assertEquals('Cannot query field "code" on type "Beam".', $response['error']);
+        $this->assertEquals('Cannot query field "code" on type "BeamClaim".', $response['error']);
 
         config([
             'enjin-platform.auth' => null,