diff --git a/tests/support.rs b/tests/support.rs
index 6ed7c71..fc5dbab 100644
--- a/tests/support.rs
+++ b/tests/support.rs
@@ -8,8 +8,9 @@ pub fn random_name() -> Name {
 pub fn setup_range(count: usize) -> (TestMoreThanText, TestDocument) {
     let test_env = TestMoreThanText::new();
     let mut mtt = test_env.get_morethantext();
+    let client = mtt.client();
     let test_doc = TestDocument::new(vec![FieldType::Integer]);
-    mtt.create_document(test_doc.get_docdef()).unwrap();
+    client.create_document(test_doc.get_docdef()).unwrap();
     let mut data: Vec<Vec<i128>> = Vec::new();
     for i in 0..count {
         let holder: i128 = i.try_into().unwrap();
@@ -72,7 +73,7 @@ impl TestDocument {
                 let field: Field = rec[i].clone().into();
                 add.add_field(self.field_names[i].clone(), field);
             }
-            mtt.records(add).unwrap();
+            mtt.client().records(add).unwrap();
         }
     }
 }