diff --git a/Civi/DataProcessor/Source/Contact/MultipleCustomGroupSource.php b/Civi/DataProcessor/Source/Contact/MultipleCustomGroupSource.php
index d3e1073aa303505c32177d9e12b8569cd802611a..ac15fc7b262c5df3a64e56a5b5ac3c56dd3720e1 100644
--- a/Civi/DataProcessor/Source/Contact/MultipleCustomGroupSource.php
+++ b/Civi/DataProcessor/Source/Contact/MultipleCustomGroupSource.php
@@ -92,7 +92,6 @@ class MultipleCustomGroupSource extends AbstractSource {
    *
    * @param DataSpecification $dataSpecification
    * @param bool $onlySearchAbleFields
-   * @param $entity
    * @throws \Civi\DataProcessor\DataSpecification\FieldExistsException
    * @throws \Exception
    */
@@ -119,6 +118,7 @@ class MultipleCustomGroupSource extends AbstractSource {
    * Ensures a field is in the data source
    *
    * @param \Civi\DataProcessor\DataSpecification\FieldSpecification $fieldSpecification
+   * @return \Civi\DataProcessor\Source\SourceInterface
    * @throws \Exception
    */
   public function ensureFieldInSource(FieldSpecification $fieldSpecification) {
@@ -127,6 +127,7 @@ class MultipleCustomGroupSource extends AbstractSource {
     } catch (FieldExistsException $e) {
       // Do nothing.
     }
+    return $this;
   }