diff --git a/aws/data_source_aws_ecr_lifecycle_policy_document_test.go b/aws/data_source_aws_ecr_lifecycle_policy_document_test.go
index bb7b297f9aa..647af259df5 100644
--- a/aws/data_source_aws_ecr_lifecycle_policy_document_test.go
+++ b/aws/data_source_aws_ecr_lifecycle_policy_document_test.go
@@ -52,11 +52,13 @@ var testAccAWSEcrLifecyclePolicyDocumentConfig = `
 data "aws_ecr_lifecycle_policy_document" "test" {
     rule {
       priority        = 1
-      description     = "This is a test."
-      tag_status      = "tagged"
-      tag_prefix_list = ["prod"]
-      count_type      = "imageCountMoreThan"
-      count_number    = 100
+			description     = "This is a test."
+			selection 			= {
+				tag_status      = "tagged"
+				tag_prefix_list = ["prod"]
+				count_type      = "imageCountMoreThan"
+				count_number    = 100
+			}
     }
 }
 `
@@ -66,15 +68,17 @@ var testAccAWSEcrLifecyclePolicyDocumentExpectedJSON = `{
     {
       "rulePriority": 1,
       "description": "This is a test.",
-      "selection": [
+      "selection": {
         "tagStatus": "tagged",
-        "tagPrefixList": ["prod"],
+        "tagPrefixList": [
+          "prod"
+        ],
         "countType": "imageCountMoreThan",
         "countNumber": 100
-      ],
-      "action": [
+      },
+      "action": {
         "type": "expire"
-      ]
+      }
     }
   ]
 }`