diff --git a/cmd/image-builder/distro_test.go b/cmd/image-builder/distro_test.go
index 7c187f8..547645d 100644
--- a/cmd/image-builder/distro_test.go
+++ b/cmd/image-builder/distro_test.go
@@ -6,7 +6,7 @@ import (
 
 	"github.com/stretchr/testify/assert"
 
-	"github.com/osbuild/image-builder-cli/cmd/image-builder"
+	main "github.com/osbuild/image-builder-cli/cmd/image-builder"
 )
 
 func TestFindDistro(t *testing.T) {
diff --git a/cmd/image-builder/filters_test.go b/cmd/image-builder/filters_test.go
index 2bf984a..0d84801 100644
--- a/cmd/image-builder/filters_test.go
+++ b/cmd/image-builder/filters_test.go
@@ -7,7 +7,7 @@ import (
 
 	testrepos "github.com/osbuild/images/test/data/repositories"
 
-	"github.com/osbuild/image-builder-cli/cmd/image-builder"
+	main "github.com/osbuild/image-builder-cli/cmd/image-builder"
 )
 
 func TestGetOneImageHappy(t *testing.T) {
diff --git a/cmd/image-builder/main_test.go b/cmd/image-builder/main_test.go
index a7eb06b..fcc8bad 100644
--- a/cmd/image-builder/main_test.go
+++ b/cmd/image-builder/main_test.go
@@ -13,7 +13,7 @@ import (
 
 	testrepos "github.com/osbuild/images/test/data/repositories"
 
-	"github.com/osbuild/image-builder-cli/cmd/image-builder"
+	main "github.com/osbuild/image-builder-cli/cmd/image-builder"
 	"github.com/osbuild/image-builder-cli/internal/manifesttest"
 	"github.com/osbuild/image-builder-cli/internal/testutil"
 )