Fix test failures.

1 files changed, 2 insertions, 2 deletions

diff --git a/plugins/SegmentEditor/tests/Unit/SegmentQueryDecoratorTest.php b/plugins/SegmentEditor/tests/Unit/SegmentQueryDecoratorTest.php
index 4c86f54fbe..6ecad82f2f 100644
--- a/plugins/SegmentEditor/tests/Unit/SegmentQueryDecoratorTest.php
+++ b/plugins/SegmentEditor/tests/Unit/SegmentQueryDecoratorTest.php
@@ -82,8 +82,8 @@ class SegmentQueryDecoratorTest extends \PHPUnit_Framework_TestCase
     private function getMockSegmentEditorService()
     {
         $mock = $this->getMock('Piwik\Plugins\SegmentEditor\Services\StoredSegmentService',
-            array('getSegmentsToAutoArchive'), array(), '', $callOriginalConstructor = false);
-        $mock->expects($this->any())->method('getSegmentsToAutoArchive')->willReturn(self::$storedSegments);
+            array('getAllSegmentsAndIgnoreVisibility'), array(), '', $callOriginalConstructor = false);
+        $mock->expects($this->any())->method('getAllSegmentsAndIgnoreVisibility')->willReturn(self::$storedSegments);
         return $mock;
     }
 }