diff --git a/lib/ddtrace/contrib/aws/integration.rb b/lib/ddtrace/contrib/aws/integration.rb index e9c6ab2f42..fac9e19412 100644 --- a/lib/ddtrace/contrib/aws/integration.rb +++ b/lib/ddtrace/contrib/aws/integration.rb @@ -12,7 +12,11 @@ class Integration register_as :aws, auto_patch: true def self.version - Gem.loaded_specs['aws-sdk'] && Gem.loaded_specs['aws-sdk'].version + if Gem.loaded_specs['aws-sdk'] + Gem.loaded_specs['aws-sdk'].version + elsif Gem.loaded_specs['aws-sdk-core'] + Gem.loaded_specs['aws-sdk-core'].version + end end def self.present?