diff --git a/backend/dataall/modules/redshift_datasets_shares/services/redshift_table_share_validator.py b/backend/dataall/modules/redshift_datasets_shares/services/redshift_table_share_validator.py index f46a919ae..a0f87a439 100644 --- a/backend/dataall/modules/redshift_datasets_shares/services/redshift_table_share_validator.py +++ b/backend/dataall/modules/redshift_datasets_shares/services/redshift_table_share_validator.py @@ -48,7 +48,7 @@ def validate_share_object_create( @staticmethod def validate_share_object_submit(session, dataset, share) -> bool: - environment = EnvironmentService.get_environment_by_uri(session, dataset.environmentUri) + environment = EnvironmentService.get_environment_by_uri(session, share.environmentUri) RedshiftTableValidator._validate_redshift_role( session=session, environment=environment, @@ -60,7 +60,7 @@ def validate_share_object_submit(session, dataset, share) -> bool: @staticmethod def validate_share_object_approve(session, dataset, share) -> bool: - environment = EnvironmentService.get_environment_by_uri(session, dataset.environmentUri) + environment = EnvironmentService.get_environment_by_uri(session, share.environmentUri) RedshiftTableValidator._validate_redshift_role( session=session, environment=environment,