diff --git a/apps/contracts/src/tests/testTwoStrategiesVault.ts b/apps/contracts/src/tests/testTwoStrategiesVault.ts
index a2178caf..e2285976 100644
--- a/apps/contracts/src/tests/testTwoStrategiesVault.ts
+++ b/apps/contracts/src/tests/testTwoStrategiesVault.ts
@@ -174,13 +174,13 @@ console.log('🚀 « investResult:', investResult);
 
 // Step 3: Rebalance Vault
 const rebalanceParams: Instruction[] = [
-  // {
-  //   action: ActionType.Withdraw,
-  //   strategy: addressBook.getContractId("hodl_strategy"),
-  //   amount: BigInt(500_0_000_000),
-  //   swap_details_exact_in: undefined,
-  //   swap_details_exact_out: undefined,
-  // },
+  {
+    action: ActionType.Withdraw,
+    strategy: addressBook.getContractId("hodl_strategy"),
+    amount: BigInt(500_0_000_000),
+    swap_details_exact_in: undefined,
+    swap_details_exact_out: undefined,
+  },
   {
     action: ActionType.Invest,
     strategy: addressBook.getContractId("fixed_apr_strategy"),
diff --git a/apps/contracts/src/utils/tx.ts b/apps/contracts/src/utils/tx.ts
index c841e0df..08074e61 100644
--- a/apps/contracts/src/utils/tx.ts
+++ b/apps/contracts/src/utils/tx.ts
@@ -81,7 +81,7 @@ export async function invokeTransaction(
   const prepped_tx = assemble_tx.setSorobanData(sim_tx_data).build();
   prepped_tx.sign(source);
   const tx_hash = prepped_tx.hash().toString("hex");
-  console.log('🛑🛑🛑🛑🛑:', prepped_tx.toXDR());
+  console.log('🛑 SIGNED TX:', prepped_tx.toXDR());
 
   console.log("submitting tx...");
   let response: txResponse = await loadedConfig.rpc.sendTransaction(prepped_tx);
diff --git a/apps/contracts/vault/src/lib.rs b/apps/contracts/vault/src/lib.rs
index 77d1a55b..e226ad8e 100755
--- a/apps/contracts/vault/src/lib.rs
+++ b/apps/contracts/vault/src/lib.rs
@@ -757,8 +757,9 @@ impl VaultManagementTrait for DeFindexVault {
                 },
                 ActionType::Invest => match (&instruction.strategy, &instruction.amount) {
                     (Some(strategy_address), Some(amount)) => {
+                        let asset_address = get_strategy_asset(&e, strategy_address)?;
                         invest_in_strategy(
-                            &e, strategy_address, strategy_address, amount)?; // TODO THIS WILL FAIUL FOR NOW
+                            &e, &asset_address.address, strategy_address, amount)?;
                     }
                     _ => return Err(ContractError::MissingInstructionData),
                 },