diff --git a/tests/integration/testdata/sync/infra/after/Ruby/function/app.rb b/tests/integration/testdata/sync/infra/after/Ruby/function/app.rb
index e261c6e9a0..c0875db775 100644
--- a/tests/integration/testdata/sync/infra/after/Ruby/function/app.rb
+++ b/tests/integration/testdata/sync/infra/after/Ruby/function/app.rb
@@ -1,11 +1,11 @@
-require 'statistics'
+require 'ruby-statistics/distribution/normal'
 require 'json'
 require 'layer'
 
 def lambda_handler(event:, context:)
   # Sample pure Lambda function that returns a message and a location
 
-  normal = Statistics::Distribution::Normal.new(2,3)
+  normal = RubyStatistics::Distribution::Normal.new(2,3)
 
   {
     statusCode: 200,
diff --git a/tests/integration/testdata/sync/infra/before/Ruby/function/app.rb b/tests/integration/testdata/sync/infra/before/Ruby/function/app.rb
index c4699fe6ea..6d20b3e9c1 100644
--- a/tests/integration/testdata/sync/infra/before/Ruby/function/app.rb
+++ b/tests/integration/testdata/sync/infra/before/Ruby/function/app.rb
@@ -1,11 +1,11 @@
-require 'statistics'
+require 'ruby-statistics/distribution/normal'
 require 'json'
 require 'layer'
 
 def lambda_handler(event:, context:)
   # Sample pure Lambda function that returns a message and a location
 
-  normal = Statistics::Distribution::Normal.new(2,3)
+  normal = RubyStatistics::Distribution::Normal.new(2,3)
 
   {
     statusCode: 200,