Update instrumentation/aws_lambda/lib/opentelemetry/instrumentation/aws_lambda/wrap.rb

Co-authored-by: Robert <[email protected]>

Author: xuan-cao-swi

instrumentation/aws_lambda/lib/opentelemetry/instrumentation/aws_lambda/wrap.rb

@@ -43,7 +43,7 @@ def wrap_lambda(event:, context:, handler:, flush_timeout: DEFAULT_FLUSH_TIMEOUT
               begin
                 response = yield
 
-                if span.respond_to?(:attributes) && !span.attributes.key?(OpenTelemetry::SemanticConventions::Trace::HTTP_STATUS_CODE)
+                if span.recording? && !span.attributes.key?(OpenTelemetry::SemanticConventions::Trace::HTTP_STATUS_CODE)
                   status_code = response['statusCode'] || response[:statusCode] if response.is_a?(Hash)
                   span.set_attribute(OpenTelemetry::SemanticConventions::Trace::HTTP_STATUS_CODE, status_code) if status_code
                 end