diff --git a/crates/iota-genesis-builder/src/stardust/migration/tests/executor.rs b/crates/iota-genesis-builder/src/stardust/migration/tests/executor.rs
index a5eb345425f..ebf913a48e7 100644
--- a/crates/iota-genesis-builder/src/stardust/migration/tests/executor.rs
+++ b/crates/iota-genesis-builder/src/stardust/migration/tests/executor.rs
@@ -38,7 +38,7 @@ fn create_bag_with_pt() {
         .with_unlock_conditions([UnlockCondition::from(
             ImmutableAliasAddressUnlockCondition::new(owner),
         )])
-        .finish_with_params(supply)
+        .finish()
         .unwrap();
     let foundry_id = foundry.id();
     let foundry_package_data = NativeTokenPackageData::new(
diff --git a/crates/iota-genesis-builder/src/stardust/test_outputs/alias_ownership.rs b/crates/iota-genesis-builder/src/stardust/test_outputs/alias_ownership.rs
index 00bf2213725..26d455c8f2b 100644
--- a/crates/iota-genesis-builder/src/stardust/test_outputs/alias_ownership.rs
+++ b/crates/iota-genesis-builder/src/stardust/test_outputs/alias_ownership.rs
@@ -166,15 +166,15 @@ fn random_foundry_output(
 ) -> (OutputHeader, FoundryOutput) {
     let foundry_output_header = random_output_header(rng);
 
+    let amount = rng.gen_range(1_000_000..10_000_000);
     let supply = rng.gen_range(1_000_000..100_000_000);
-    let amount = rng.gen_range(1_000_000..10_000_000).min(supply);
     let token_scheme = SimpleTokenScheme::new(supply, 0, supply).unwrap();
     let foundry_output =
         FoundryOutputBuilder::new_with_amount(amount, *serial_number, token_scheme.into())
             .with_unlock_conditions([UnlockCondition::from(
                 ImmutableAliasAddressUnlockCondition::new(owner),
             )])
-            .finish_with_params(supply)
+            .finish()
             .unwrap();
 
     *serial_number += 1;