diff --git a/build_tools/aws-sdk-code-generator/lib/aws-sdk-code-generator/views/types_module.rb b/build_tools/aws-sdk-code-generator/lib/aws-sdk-code-generator/views/types_module.rb
index 981665c9cfa..492c9e76faa 100644
--- a/build_tools/aws-sdk-code-generator/lib/aws-sdk-code-generator/views/types_module.rb
+++ b/build_tools/aws-sdk-code-generator/lib/aws-sdk-code-generator/views/types_module.rb
@@ -92,7 +92,7 @@ def struct_members(shape)
           case member_target["type"]
           when 'map'
             key_shape = @api['shapes'][member_target['key']['shape']]
-            value_shape = @api['shapes'][member_target['key']['shape']]
+            value_shape = @api['shapes'][member_target['value']['shape']]
             sensitive ||= !!(key_shape['sensitive'] || value_shape['sensitive'])
           when 'list'
             list_member = @api['shapes'][member_target['member']['shape']]