diff --git a/dkan_harvest.migrate.inc b/dkan_harvest.migrate.inc
index fb6b686..128f4d8 100644
--- a/dkan_harvest.migrate.inc
+++ b/dkan_harvest.migrate.inc
@@ -1113,13 +1113,6 @@ class HarvestMigration extends MigrateDKAN {
     return $group_node;
   }
 
-  /**
-   * Getter for harvest source. This is mainly used for unittests.
-   */
-  public function getHarvestSource() {
-    return $this->dkanHarvestSource;
-  }
-
   /**
    * Getter for the $logID variable. Mainly used for the HarvestMigrateSQLMap
    * class.
diff --git a/test/phpunit/DkanHarvestTest.php b/test/phpunit/DkanHarvestTest.php
index 659e90f..a80d36b 100644
--- a/test/phpunit/DkanHarvestTest.php
+++ b/test/phpunit/DkanHarvestTest.php
@@ -84,7 +84,7 @@ public function test_dkan_harvest_get_migration() {
     // Get first migration registration.
     $harvest_test_source_local_dir = $sources['harvest_test_source_local_dir'];
     $migration_harvest_test_source_local_dir = dkan_harvest_get_migration($harvest_test_source_local_dir);
-    $this->assertEquals($harvest_test_source_local_dir, $migration_harvest_test_source_local_dir->getHarvestSource());
+    $this->assertEquals($harvest_test_source_local_dir, $migration_harvest_test_source_local_dir->getSource());
 
     // If we change the source (keeping the machine name) the migration should
     // have the updated harvest source.
@@ -92,7 +92,7 @@ public function test_dkan_harvest_get_migration() {
     $harvest_test_source_local_dir->uri = DRUPAL_ROOT . "/" . drupal_get_path('module',
       'dkan_harvest') . "/test/phpunit/data/harvest_fictional_test_source/";
     $migration_harvest_test_source_second = dkan_harvest_get_migration($harvest_test_source_local_dir);
-    $this->assertEquals($harvest_test_source_local_dir, $migration_harvest_test_source_second->getHarvestSource());
+    $this->assertEquals($harvest_test_source_local_dir, $migration_harvest_test_source_second->getSource());
   }
 
   /**