e2e: fix TestFormContract

SiaFoundation · Aug 22, 2024 · 49d7f66 · 49d7f66
1 parent 7fde826
commit 49d7f66
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal/test/e2e/contracts_test.go b/internal/test/e2e/contracts_test.go
@@ -35,10 +35,10 @@ func TestFormContract(t *testing.T) {
 	tt.OK(err)
 
 	// form a contract using the bus
-	cs, _ := b.ConsensusState(context.Background())
 	wallet, _ := b.Wallet(context.Background())
-	endHeight := cs.BlockHeight + test.AutopilotConfig.Contracts.Period + test.AutopilotConfig.Contracts.RenewWindow
-	contract, err := b.FormContract(context.Background(), wallet.Address, types.Siacoins(1), h.PublicKey, h.NetAddress, types.Siacoins(1), endHeight)
+	ap, err := b.Autopilot(context.Background(), api.DefaultAutopilotID)
+	tt.OK(err)
+	contract, err := b.FormContract(context.Background(), wallet.Address, types.Siacoins(1), h.PublicKey, h.NetAddress, types.Siacoins(1), ap.EndHeight())
 	tt.OK(err)
 
 	// assert the contract was added to the bus