diff --git a/templates/terraform.tf.tpl b/templates/terraform.tf.tpl
index b6792d8..5ad0e94 100644
--- a/templates/terraform.tf.tpl
+++ b/templates/terraform.tf.tpl
@@ -8,8 +8,7 @@ terraform {
     profile = "${profile}"
     encrypt = "${encrypt}"
     %{~ if role_arn != "" ~}
-
-    assume_role {
+    assume_role = {
       role_arn = "${role_arn}"
     }
     %{~ endif ~}
diff --git a/variables.tf b/variables.tf
index 613c898..a2c4153 100644
--- a/variables.tf
+++ b/variables.tf
@@ -104,7 +104,7 @@ variable "profile" {
 variable "role_arn" {
   type        = string
   default     = ""
-  description = "The role to be assumed"
+  description = "The role to be assumed. If this is set, `terraform_version` variable must be `1.6.0` or higher."
 }
 
 variable "terraform_backend_config_file_name" {