diff --git a/internal/controller/helpers_test.go b/internal/controller/helpers_test.go
index c4a12bf4d0..afdcee2c1f 100644
--- a/internal/controller/helpers_test.go
+++ b/internal/controller/helpers_test.go
@@ -24,7 +24,7 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"sigs.k8s.io/cluster-api/util"
 
-	infrav1 "github.com/nutanix-cloud-native/cluster-api-provider-nutanix/api/v1beta1"
+	infrav1 "github.com/nutanix-cloud-native/cluster-api-provider-nutanix/api/infrastructure/v1beta1"
 
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
diff --git a/test/e2e/failure_domains_test.go b/test/e2e/failure_domains_test.go
index 107d37676b..7f7db3e245 100644
--- a/test/e2e/failure_domains_test.go
+++ b/test/e2e/failure_domains_test.go
@@ -28,7 +28,7 @@ import (
 	clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
 	"sigs.k8s.io/cluster-api/test/framework/clusterctl"
 
-	infrav1 "github.com/nutanix-cloud-native/cluster-api-provider-nutanix/api/v1beta1"
+	infrav1 "github.com/nutanix-cloud-native/cluster-api-provider-nutanix/api/infrastructure/v1beta1"
 )
 
 const (