diff --git a/scripts/swap_exact_input_script/tests/cases/success.rs b/scripts/swap_exact_input_script/tests/cases/success.rs
index 4dfd89a..295b4a5 100644
--- a/scripts/swap_exact_input_script/tests/cases/success.rs
+++ b/scripts/swap_exact_input_script/tests/cases/success.rs
@@ -1,5 +1,5 @@
+use fuels::prelude::VariableOutputPolicy;
 use crate::utils::setup;
-use fuels::programs::call_utils::TxDependencyExtension;
 use test_harness::interface::amm::pool_metadata;
 use test_harness::interface::scripts::get_transaction_inputs_outputs;
 use test_harness::utils::common::{pool_assets_balance, MINIMUM_LIQUIDITY};
@@ -41,7 +41,7 @@ async fn swap_between_two_volatile_tokens() {
         .with_contracts(&[&amm.instance])
         .with_inputs(inputs)
         .with_outputs(outputs)
-        .append_variable_outputs(2)
+        .with_variable_output_policy(VariableOutputPolicy::Exactly(2))
         .call()
         .await
         .unwrap()
@@ -72,7 +72,7 @@ async fn swap_between_two_volatile_tokens() {
         .with_contracts(&[&amm.instance])
         .with_inputs(inputs)
         .with_outputs(outputs)
-        .append_variable_outputs(1)
+        .with_variable_output_policy(VariableOutputPolicy::Exactly(1))
         .call()
         .await
         .unwrap()
@@ -137,7 +137,7 @@ async fn swap_between_three_volatile_tokens() {
         .with_contracts(&[&amm.instance])
         .with_inputs(inputs)
         .with_outputs(outputs)
-        .append_variable_outputs(2)
+        .with_variable_output_policy(VariableOutputPolicy::Exactly(2))
         .call()
         .await
         .unwrap()
@@ -165,7 +165,7 @@ async fn swap_between_three_volatile_tokens() {
         .with_contracts(&[&amm.instance])
         .with_inputs(inputs)
         .with_outputs(outputs)
-        .append_variable_outputs(2)
+        .with_variable_output_policy(VariableOutputPolicy::Exactly(2))
         .call()
         .await
         .unwrap()
@@ -199,7 +199,7 @@ async fn swap_between_three_volatile_tokens() {
         .with_contracts(&[&amm.instance])
         .with_inputs(inputs)
         .with_outputs(outputs)
-        .append_variable_outputs(1)
+        .with_variable_output_policy(VariableOutputPolicy::Exactly(1))
         .call()
         .await
         .unwrap()
diff --git a/test-harness/src/interface.rs b/test-harness/src/interface.rs
index d5d7abb..d82f2f3 100644
--- a/test-harness/src/interface.rs
+++ b/test-harness/src/interface.rs
@@ -38,7 +38,7 @@ pub mod amm {
     pub async fn initialize_ownership(
         contract: &MiraAMM<WalletUnlocked>,
         owner: Identity,
-    ) -> FuelCallResponse<()> {
+    ) -> CallResponse<()> {
         contract
             .methods()
             .transfer_ownership(owner)