terraform-provider-elasticsearch: Regression issue on 1.5.1

After upgrading to 1.5.1 I am getting the following error:

Error: health check timeout: Head "https://elasticsearch.mydomain.com": RequestCanceled: request context canceled
caused by: context deadline exceeded: no Elasticsearch node available

It could be related to aws_assume_role_arn as I use it on my provider config:

provider "elasticsearch" {
  url                 = "https://elasticsearch.mydomain.com"
  aws_region          = "eu-west-1"
  aws_profile         = ""
  aws_assume_role_arn = "arn:aws:iam::111111111:role/Role"
  sign_aws_requests   = true
}

It only seems to happen if I use aws_assume_role_arn and it does not when I use aws_profile. I am using Elasticsearch 7.9.

Reverting back to 1.5.0 and the error disappears.

I see there’s significant changes done in this PR https://github.com/phillbaker/terraform-provider-elasticsearch/pull/119, maybe it’s related.

Thanks.