diff --git a/influxdb2_client/src/api/setup.rs b/influxdb2_client/src/api/setup.rs
index 4f547fe89e..87b99ee5bb 100644
--- a/influxdb2_client/src/api/setup.rs
+++ b/influxdb2_client/src/api/setup.rs
@@ -156,7 +156,7 @@ mod tests {
         let client = Client::new(&mockito::server_url(), token);
 
         let _result = client
-            .setup_init(username, org, bucket, password, retention_period_hrs)
+            .setup_init(username, org, bucket, Some(password.to_string()), Some(retention_period_hrs), None)
             .await;
 
         mock_server.assert();
@@ -185,7 +185,7 @@ mod tests {
         let client = Client::new(&mockito::server_url(), token);
 
         let _result = client
-            .setup_new(username, org, bucket, password, retention_period_hrs)
+            .setup_new(username, org, bucket, Some(password.to_string()), Some(retention_period_hrs), None)
             .await;
 
         mock_server.assert();