diff --git a/azurerm/internal/services/analysisservices/validate/querypool_connection_mode.go b/azurerm/internal/services/analysisservices/validate/querypool_connection_mode.go
index a93c1bb78804..9ca44201d8fc 100644
--- a/azurerm/internal/services/analysisservices/validate/querypool_connection_mode.go
+++ b/azurerm/internal/services/analysisservices/validate/querypool_connection_mode.go
@@ -2,11 +2,11 @@ package validate
 
 import (
 	"github.com/Azure/azure-sdk-for-go/services/analysisservices/mgmt/2017-08-01/analysisservices"
-	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
 	"github.com/hashicorp/terraform-plugin-sdk/helper/validation"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/tf/pluginsdk"
 )
 
-func QueryPoolConnectionMode() schema.SchemaValidateFunc {
+func QueryPoolConnectionMode() pluginsdk.SchemaValidateFunc {
 	connectionModes := make([]string, len(analysisservices.PossibleConnectionModeValues()))
 	for i, v := range analysisservices.PossibleConnectionModeValues() {
 		connectionModes[i] = string(v)