diff --git a/internal/service/auditmanager/control_data_source.go b/internal/service/auditmanager/control_data_source.go
index 35ef031828d..4497b30b645 100644
--- a/internal/service/auditmanager/control_data_source.go
+++ b/internal/service/auditmanager/control_data_source.go
@@ -25,7 +25,7 @@ import (
 	"github.com/hashicorp/terraform-provider-aws/names"
 )
 
-// @FrameworkDataSource
+// @FrameworkDataSource("aws_auditmanager_control", name="Control")
 func newDataSourceControl(context.Context) (datasource.DataSourceWithConfigure, error) {
 	return &dataSourceControl{}, nil
 }
diff --git a/internal/service/auditmanager/framework_data_source.go b/internal/service/auditmanager/framework_data_source.go
index 815a778beb9..da10ab66e60 100644
--- a/internal/service/auditmanager/framework_data_source.go
+++ b/internal/service/auditmanager/framework_data_source.go
@@ -23,7 +23,7 @@ import (
 	"github.com/hashicorp/terraform-provider-aws/names"
 )
 
-// @FrameworkDataSource
+// @FrameworkDataSource("aws_auditmanager_framework", name="Framework")
 func newDataSourceFramework(context.Context) (datasource.DataSourceWithConfigure, error) {
 	return &dataSourceFramework{}, nil
 }
diff --git a/internal/service/auditmanager/service_package_gen.go b/internal/service/auditmanager/service_package_gen.go
index b134a2d9fc1..95783acd5a5 100644
--- a/internal/service/auditmanager/service_package_gen.go
+++ b/internal/service/auditmanager/service_package_gen.go
@@ -18,9 +18,11 @@ func (p *servicePackage) FrameworkDataSources(ctx context.Context) []*types.Serv
 	return []*types.ServicePackageFrameworkDataSource{
 		{
 			Factory: newDataSourceControl,
+			Name:    "Control",
 		},
 		{
 			Factory: newDataSourceFramework,
+			Name:    "Framework",
 		},
 	}
 }