diff --git a/spec/datadog/di/probe_builder_spec.rb b/spec/datadog/di/probe_builder_spec.rb
index 0a8e7609bd3..f81c0f34ff3 100644
--- a/spec/datadog/di/probe_builder_spec.rb
+++ b/spec/datadog/di/probe_builder_spec.rb
@@ -1,6 +1,9 @@
+require "datadog/di/spec_helper"
 require "datadog/di/probe_builder"
 
 RSpec.describe Datadog::DI::ProbeBuilder do
+  di_test
+
   describe ".build_from_remote_config" do
     let(:probe) do
       described_class.build_from_remote_config(rc_probe_spec)
diff --git a/spec/datadog/di/probe_spec.rb b/spec/datadog/di/probe_spec.rb
index e37832ac46e..e0c78aa0723 100644
--- a/spec/datadog/di/probe_spec.rb
+++ b/spec/datadog/di/probe_spec.rb
@@ -1,6 +1,9 @@
+require "datadog/di/spec_helper"
 require "datadog/di/probe"
 
 RSpec.describe Datadog::DI::Probe do
+  di_test
+
   shared_context "method probe" do
     let(:probe) do
       described_class.new(id: "42", type: "foo", type_name: "Foo", method_name: "bar")