diff --git a/tests/modules/media_entity_test_type/src/Plugin/MediaEntity/Type/TestType.php b/tests/modules/media_entity_test_type/src/Plugin/MediaEntity/Type/TestType.php
index bae89fe..d34114d 100644
--- a/tests/modules/media_entity_test_type/src/Plugin/MediaEntity/Type/TestType.php
+++ b/tests/modules/media_entity_test_type/src/Plugin/MediaEntity/Type/TestType.php
@@ -64,7 +64,7 @@ class TestType extends Generic {
       '#default_value' => empty($this->configuration['test_config_value']) ? NULL : $this->configuration['test_config_value'],
     ];
 
-    $form = $this->defaultFieldConfigurationForm($form, $form_state);
+    $form = $this->sourceFieldConfigurationForm($form, $form_state);
 
     $form['source_field']['#description'] = $this->t('Field on media entity that stores the source information.');