diff --git a/dns_zones.tf b/dns_zones.tf
index 19cf7f2..fc76398 100644
--- a/dns_zones.tf
+++ b/dns_zones.tf
@@ -30,7 +30,7 @@ provider "aws" {
 provider "aws" {
   alias = "self"
   assume_role {
-    role_arn     = format("arn:%v:iam::%v:role/r-inf-terraform", data.aws_arn.current.partition, data.aws_arn.current.account_id)
+    role_arn     = format("arn:%v:iam::%v:role/r-inf-terraform", data.aws_arn.current.partition, data.aws_caller_identity.current.account_id)
     session_name = var.os_username
   }
 }