diff --git a/chainio/consumer_test.go b/chainio/consumer_test.go
index 3ef79b61b4b..d1cabf3168e 100644
--- a/chainio/consumer_test.go
+++ b/chainio/consumer_test.go
@@ -4,7 +4,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/lightningnetwork/lnd/fn"
+	"github.com/lightningnetwork/lnd/fn/v2"
 	"github.com/stretchr/testify/require"
 )
 
diff --git a/chainio/dispatcher_test.go b/chainio/dispatcher_test.go
index 88044c0201d..11abbeb65e0 100644
--- a/chainio/dispatcher_test.go
+++ b/chainio/dispatcher_test.go
@@ -5,7 +5,7 @@ import (
 	"time"
 
 	"github.com/lightningnetwork/lnd/chainntnfs"
-	"github.com/lightningnetwork/lnd/fn"
+	"github.com/lightningnetwork/lnd/fn/v2"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/require"
 )