1 files changed, 4 insertions, 2 deletions
diff --git a/tests/unit/unittest/mockfilepathstorage.h b/tests/unit/unittest/mockfilepathstorage.h
index b1114747b1..7056ee2480 100644
--- a/tests/unit/unittest/mockfilepathstorage.h
+++ b/tests/unit/unittest/mockfilepathstorage.h
@@ -27,6 +27,8 @@
 
 #include "googletest.h"
 
+#include <filepathstoragesources.h>
+
 class MockFilePathStorage
 {
 public:
@@ -36,7 +38,7 @@ public:
                  int (int directoryId, Utils::SmallStringView sourceName));
     MOCK_METHOD1(fetchDirectoryPath,
                  Utils::PathString (int directoryId));
-    MOCK_METHOD1(fetchSourceName,
-                 Utils::SmallString (int sourceId));
+    MOCK_METHOD1(fetchSourceNameAndDirectoryId,
+                 ClangBackEnd::Sources::SourceNameAndDirectoryId (int sourceId));
 };