diff --git a/createtx_test.go b/createtx_test.go
index d1d3d6d..379ff2a 100644
--- a/createtx_test.go
+++ b/createtx_test.go
@@ -28,7 +28,7 @@ func TestFakeTxs(t *testing.T) {
 	// This will pass validation because btcscript is unaware of invalid
 	// tx inputs, however, this example would fail in btcd.
 	utxo := &tx.Utxo{}
-	addr, err := w.NextUnusedAddress()
+	addr, err := w.NextChainedAddress(&wallet.BlockStamp{})
 	if err != nil {
 		t.Errorf("Cannot get next address: %s", err)
 		return
diff --git a/wallet/wallet_test.go b/wallet/wallet_test.go
index 7cf530b..f5c9639 100644
--- a/wallet/wallet_test.go
+++ b/wallet/wallet_test.go
@@ -42,7 +42,7 @@ func TestBtcAddressSerializer(t *testing.T) {
 		t.Error(err.Error())
 		return
 	}
-	addr, err := newBtcAddress(privKey, nil, &BlockStamp{})
+	addr, err := newBtcAddress(privKey, nil, &BlockStamp{}, true)
 	if err != nil {
 		t.Error(err.Error())
 		return