*vc-diff*.diff

diff --git a/.coveragerc b/.coveragerc
index 612ec54..2425b39 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -1,6 +1,6 @@
 [run]
 branch = True
-source = vumi
+source = vumi_message_store
 
 [report]
 exclude_lines = 
diff --git a/vumi_message_store/message_store.py b/vumi_message_store/message_store.py
index 98a1fb4..bbf5975 100644
--- a/vumi_message_store/message_store.py
+++ b/vumi_message_store/message_store.py
@@ -1,4 +1,4 @@
-# -*- test-case-name: vumi.components.tests.test_message_store -*-
+# -*- test-case-name: vumi_message_store.tests.test_message_store -*-
 
 """Message store."""
 
diff --git a/vumi_message_store/tests/old_models.py b/vumi_message_store/tests/old_models.py
index f6a0e78..58b307e 100644
--- a/vumi_message_store/tests/old_models.py
+++ b/vumi_message_store/tests/old_models.py
@@ -5,7 +5,7 @@ from vumi.message import TransportUserMessage, TransportEvent
 from vumi.persist.model import Model
 from vumi.persist.fields import (
     VumiMessage, ForeignKey, ListOf, Dynamic, Tag, Unicode, ManyToMany)
-from vumi.components.message_store_migrators import (
+from vumi_message_store.migrators import (
     InboundMessageMigrator, OutboundMessageMigrator)
 
 

👍