diff --git a/ssb/tests/test_feed.py b/ssb/tests/test_feed.py
index 34e8b40..9a06a47 100644
--- a/ssb/tests/test_feed.py
+++ b/ssb/tests/test_feed.py
@@ -3,7 +3,8 @@ from collections import OrderedDict
 
 import pytest
 
-from ssb.keys import KeyPair, Message
+from ssb.feed import Message
+from ssb.keys import KeyPair
 
 
 @pytest.fixture()