diff --git a/tests/modules/token_module_test/token_module_test.module b/tests/modules/token_module_test/token_module_test.module
index 9832f39..7caf121 100644
--- a/tests/modules/token_module_test/token_module_test.module
+++ b/tests/modules/token_module_test/token_module_test.module
@@ -24,5 +24,8 @@ function token_module_test_page_attachments() {
  */
 function token_module_test_node_presave(NodeInterface $node) {
   // Transform tokens in the body.
-  $node->body->value = \Drupal::token()->replace($node->body->value, ['node' => $node]);
+  if ($node->hasField('body')) {
+    $node->body->value = \Drupal::token()
+      ->replace($node->body->value, ['node' => $node]);
+  }
 }