diff --git a/tests/src/com/android/email/mail/StoreTests.java b/tests/src/com/android/email/mail/StoreTests.java
index 0d08f98..de5d764 100644
--- a/tests/src/com/android/email/mail/StoreTests.java
+++ b/tests/src/com/android/email/mail/StoreTests.java
@@ -70,7 +70,7 @@
 
         testStore = Store.getInstance(testAccount, getContext());
         assertEquals(1, Store.sStores.size());
-        assertSame(testStore, Store.sStores.get(testAccount.mId));
+        assertSame(testStore, Store.sStores.get(testAccount.mHostAuthRecv));
         Store.sStores.clear();
 
         // IMAP
@@ -82,7 +82,7 @@
         testAccount.save(mMockContext);
         testStore = Store.getInstance(testAccount, getContext());
         assertEquals(1, Store.sStores.size());
-        assertSame(testStore, Store.sStores.get(testAccount.mId));
+        assertSame(testStore, Store.sStores.get(testAccount.mHostAuthRecv));
         Store.sStores.clear();
 
         // Unknown