diff --git a/src/Plugin/Field/FieldFormatter/CollectDataFormatter.php b/src/Plugin/Field/FieldFormatter/CollectDataFormatter.php
index a627793..490e0c2 100644
--- a/src/Plugin/Field/FieldFormatter/CollectDataFormatter.php
+++ b/src/Plugin/Field/FieldFormatter/CollectDataFormatter.php
@@ -39,7 +39,7 @@ class CollectDataFormatter extends FormatterBase implements ContainerFactoryPlug
   /**
    * The injected model plugin manager.
    *
-   * @var \Drupal\collect\ModelManagerInterface
+   * @var \Drupal\collect\Model\ModelManagerInterface
    */
   protected $modelManager;
 
@@ -83,7 +83,7 @@ class CollectDataFormatter extends FormatterBase implements ContainerFactoryPlug
     $elements = array();
 
     foreach ($items as $delta => $item) {
-      // Use a schema plugin to render data.
+      // Use a model plugin to render data.
       $model_plugin = $this->modelManager->createInstanceFromUri($item->getEntity()->getSchemaUri());
       if ($model_plugin instanceof DefaultModelPlugin) {
         $elements[$delta]['no_display'] = [
@@ -99,8 +99,7 @@ class CollectDataFormatter extends FormatterBase implements ContainerFactoryPlug
         '#type' => 'item',
       );
       if ($model_plugin instanceof SpecializedDisplayModelPluginInterface) {
-        // If schema plugin has custom build method, use it to build the
-        // output.
+        // If model plugin has custom build method, use it to build the output.
         $elements[$delta]['plugin']['data'] = $model_plugin->build($model_plugin->parse($item->getEntity()), $item->getEntity());
         return $elements;
       }