diff --git a/cmd/cli_test.go b/cmd/cli_test.go
index 4730aa0..7e25649 100644
--- a/cmd/cli_test.go
+++ b/cmd/cli_test.go
@@ -6,7 +6,7 @@ import (
 	"testing"
 	"time"
 
-	"git.froth.zone/sam/awl/cli"
+	cli "git.froth.zone/sam/awl/cmd"
 	"gotest.tools/v3/assert"
 )
 
diff --git a/cmd/dig_test.go b/cmd/dig_test.go
index 2b5b94f..6ba9cf5 100644
--- a/cmd/dig_test.go
+++ b/cmd/dig_test.go
@@ -5,7 +5,7 @@ package cli_test
 import (
 	"testing"
 
-	"git.froth.zone/sam/awl/cli"
+	cli "git.froth.zone/sam/awl/cmd"
 	"git.froth.zone/sam/awl/pkg/util"
 	"gotest.tools/v3/assert"
 )
diff --git a/cmd/misc_test.go b/cmd/misc_test.go
index d2f9a09..fa20dbe 100644
--- a/cmd/misc_test.go
+++ b/cmd/misc_test.go
@@ -5,7 +5,7 @@ package cli_test
 import (
 	"testing"
 
-	"git.froth.zone/sam/awl/cli"
+	cli "git.froth.zone/sam/awl/cmd"
 	"git.froth.zone/sam/awl/pkg/util"
 	"github.com/miekg/dns"
 	"gotest.tools/v3/assert"