1 files changed, 3 insertions, 3 deletions
diff --git a/tests/db/test_message.py b/tests/db/test_message.py
index cd50ab45..8fa7692d 100644
--- a/tests/db/test_message.py
+++ b/tests/db/test_message.py
@@ -48,15 +48,15 @@ class MockNotmuchMessage(object):
     def get_date(self):
         return self.mock_date
 
-    def get_filename(self):
-        return self.mock_filename
-
     def get_tags(self):
         return self.mock_tags
 
     def get_properties(self, prop, exact=False):
         return []
 
+    def get_filenames(self):
+        return ['/foo']
+
 
 class TestMessage(unittest.TestCase):