diff --git a/services/commtest/tests/backup/pull_backup.rs b/services/commtest/tests/backup/pull_backup.rs
--- a/services/commtest/tests/backup/pull_backup.rs
+++ b/services/commtest/tests/backup/pull_backup.rs
@@ -7,6 +7,7 @@
 
 use crate::backup_utils::{
   proto::pull_backup_response::Data, proto::pull_backup_response::Data::*,
+  proto::pull_backup_response::Identity, proto::pull_backup_response::Identity::*,
   proto::PullBackupRequest, BackupServiceClient,
 };
 
@@ -45,15 +46,26 @@
   let mut current_id: String = String::new();
   while let Some(response) = inbound.message().await? {
     let response_data: Option<Data> = response.data;
-    let id = response.id;
+    let identity: Option<Identity> = response.identity;
+    let mut backup_id: Option<String> = None;
+    let mut log_id: Option<String> = None;
+    match identity {
+      Some(BackupId(bid)) => {
+        backup_id = Some(bid)
+      },
+      Some(LogId(lid)) => {
+        log_id = Some(lid)
+      },
+      None => {},
+    };
     match response_data {
       Some(CompactionChunk(chunk)) => {
-        assert!(
-          state == State::Compaction,
+        assert_eq!(
+          state, State::Compaction,
           "invalid state, expected compaction, got {:?}",
           state
         );
-        current_id = id;
+        current_id = backup_id.expect("backup id expected but not received");
         println!(
           "compaction (id {}), pushing chunk (size: {})",
           current_id,
@@ -65,12 +77,13 @@
         if state == State::Compaction {
           state = State::Log;
         }
-        assert!(state == State::Log, "invalid state, expected compaction");
-        if id != current_id {
+        assert_eq!(state, State::Log, "invalid state, expected compaction");
+        let log_id = log_id.expect("log id expected but not received");
+        if log_id != current_id {
           result
             .log_items
-            .push(Item::new(id.clone(), Vec::new(), Vec::new()));
-          current_id = id.clone();
+            .push(Item::new(log_id.clone(), Vec::new(), Vec::new()));
+          current_id = log_id;
         }
         let log_items_size = result.log_items.len() - 1;
         result.log_items[log_items_size]