diff --git a/kolibri_explore_plugin/test/test_collectionviews.py b/kolibri_explore_plugin/test/test_collectionviews.py
index fe846aab..ad4466f4 100644
--- a/kolibri_explore_plugin/test/test_collectionviews.py
+++ b/kolibri_explore_plugin/test/test_collectionviews.py
@@ -116,8 +116,8 @@ def test_get_should_resume():
 
     # Intermediate state.
     current_state = {
-        "name": "artist",
-        "sequence": "0001",
+        "collection_name": "artist",
+        "collection_sequence": 1,
         "stage": collectionviews.DownloadStage.IMPORTING_CONTENT.name,
     }
     session["COLLECTIONS_STATE"] = current_state
@@ -133,8 +133,8 @@ def test_get_should_resume():
 
     # Completed state.
     current_state = {
-        "name": "artist",
-        "sequence": "0001",
+        "collection_name": "artist",
+        "collection_sequence": 1,
         "stage": collectionviews.DownloadStage.COMPLETED.name,
     }
     session["COLLECTIONS_STATE"] = current_state