diff --git a/tests/test-media-item-mutations.php b/tests/test-media-item-mutations.php
index bb7c2de8a..e58c14a08 100644
--- a/tests/test-media-item-mutations.php
+++ b/tests/test-media-item-mutations.php
@@ -467,7 +467,7 @@ public function updateMediaItemMutation() {
 		      description
 		      mimeType
 		      author {
-		        username
+		        id
 		      }
 		    }
 		  }
@@ -590,6 +590,7 @@ public function testUpdateMediaItemMutation() {
 		 * successfully run the mutation
 		 */
 		wp_set_current_user( $this->admin );
+
 		$actual = $this->updateMediaItemMutation();
 
 		/**
@@ -613,7 +614,9 @@ public function testUpdateMediaItemMutation() {
 						'status'           => strtolower( $this->updated_status ),
 						'pingStatus'       => $this->updated_pingStatus,
 						'mimeType'         => 'image/gif',
-						'author'           => [ 'username' => 'User 196' ],
+						'author'           => [
+							'id'       => \GraphQLRelay\Relay::toGlobalId( 'user', $this->admin ),
+						],
 					],
 				],
 			],