diff --git a/internal/provider/generators/singular-data-source/schema.tmpl b/internal/provider/generators/singular-data-source/schema.tmpl
index faea02420..09d7f7d1a 100644
--- a/internal/provider/generators/singular-data-source/schema.tmpl
+++ b/internal/provider/generators/singular-data-source/schema.tmpl
@@ -5,6 +5,7 @@ package {{ .PackageName }}
 import (
 	"context"
 
+	"github.com/hashicorp/terraform-plugin-framework/provider"
 	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
 	"github.com/hashicorp/terraform-plugin-framework/types"
 	. "github.com/hashicorp/terraform-provider-awscc/internal/generic"
@@ -17,7 +18,7 @@ func init() {
 
 // {{ .FactoryFunctionName }} returns the Terraform {{ .TerraformTypeName }} data source type.
 // This Terraform data source type corresponds to the CloudFormation {{ .CloudFormationTypeName }} resource type.
-func {{ .FactoryFunctionName }}(ctx context.Context) (tfsdk.DataSourceType, error) {
+func {{ .FactoryFunctionName }}(ctx context.Context) (provider.DataSourceType, error) {
 	attributes := {{ .RootPropertiesSchema }}
 
 	attributes["id"] = tfsdk.Attribute{