diff --git a/sdk/tests/client/input_selection/timelock.rs b/sdk/tests/client/input_selection/timelock.rs
index 09d7d11fd3..06e4227f10 100644
--- a/sdk/tests/client/input_selection/timelock.rs
+++ b/sdk/tests/client/input_selection/timelock.rs
@@ -3,7 +3,7 @@
 
 use iota_sdk::{
     client::api::input_selection::{Error, InputSelection},
-    types::block::{address::Address, protocol::protocol_parameters},
+    types::block::{address::Address, protocol::protocol_parameters, slot::SlotIndex},
 };
 use pretty_assertions::assert_eq;
 
@@ -67,7 +67,7 @@ fn timelock_equal_timestamp() {
             None,
             None,
         )],
-        None,
+        Some(SlotIndex::from(200)),
     );
     let outputs = build_outputs([Basic(
         2_000_000,
@@ -84,7 +84,7 @@ fn timelock_equal_timestamp() {
         inputs.clone(),
         outputs.clone(),
         [Address::try_from_bech32(BECH32_ADDRESS_ED25519_0).unwrap()],
-        SLOT_INDEX,
+        200,
         protocol_parameters,
     )
     .select()
@@ -121,7 +121,7 @@ fn two_outputs_one_timelock_expired() {
                 None,
             ),
         ],
-        None,
+        Some(SlotIndex::from(100)),
     );
     let outputs = build_outputs([Basic(
         2_000_000,
@@ -138,7 +138,7 @@ fn two_outputs_one_timelock_expired() {
         inputs.clone(),
         outputs.clone(),
         [Address::try_from_bech32(BECH32_ADDRESS_ED25519_0).unwrap()],
-        SLOT_INDEX,
+        100,
         protocol_parameters,
     )
     .select()
@@ -176,7 +176,7 @@ fn two_outputs_one_timelocked_one_missing() {
                 None,
             ),
         ],
-        None,
+        Some(SlotIndex::from(100)),
     );
     let outputs = build_outputs([Basic(
         2_000_000,
@@ -193,7 +193,7 @@ fn two_outputs_one_timelocked_one_missing() {
         inputs.clone(),
         outputs.clone(),
         [Address::try_from_bech32(BECH32_ADDRESS_ED25519_0).unwrap()],
-        SLOT_INDEX,
+        100,
         protocol_parameters,
     )
     .select()
@@ -219,7 +219,7 @@ fn one_output_timelock_expired() {
             None,
             None,
         )],
-        None,
+        Some(SlotIndex::from(100)),
     );
     let outputs = build_outputs([Basic(
         2_000_000,
@@ -236,7 +236,7 @@ fn one_output_timelock_expired() {
         inputs.clone(),
         outputs.clone(),
         [Address::try_from_bech32(BECH32_ADDRESS_ED25519_0).unwrap()],
-        SLOT_INDEX,
+        100,
         protocol_parameters,
     )
     .select()