diff --git a/libs/client-api-test/src/test_client.rs b/libs/client-api-test/src/test_client.rs
index dd771365..fab9d3d2 100644
--- a/libs/client-api-test/src/test_client.rs
+++ b/libs/client-api-test/src/test_client.rs
@@ -236,11 +236,10 @@ impl TestClient {
       })
       .await
       .unwrap();
-    let db_doc_state = db_collab_collab_resp.encode_collab.doc_state;
     Collab::new_with_source(
       CollabOrigin::Server,
       &db_id,
-      DataSource::DocStateV1(db_doc_state.to_vec()),
+      db_collab_collab_resp.encode_collab.into(),
       vec![],
       false,
     )