diff --git a/lib/datadog_api_client.rb b/lib/datadog_api_client.rb
index aa25a33a54e7..c8043afc38c7 100644
--- a/lib/datadog_api_client.rb
+++ b/lib/datadog_api_client.rb
@@ -3,9 +3,9 @@
 
 loader = Zeitwerk::Loader.new
 loader.tag = File.basename(__FILE__, ".rb")
-loader.collapse("lib/datadog_api_client/*/models/")
-loader.collapse("lib/datadog_api_client/*/api/")
-loader.push_dir("lib/datadog_api_client/v1", namespace: DatadogAPIClient::V1)
-loader.push_dir("lib/datadog_api_client/v2", namespace: DatadogAPIClient::V2)
+loader.collapse("#{__dir__}/datadog_api_client/*/models/")
+loader.collapse("#{__dir__}/datadog_api_client/*/api/")
+loader.push_dir("#{__dir__}/datadog_api_client/v1", namespace: DatadogAPIClient::V1)
+loader.push_dir("#{__dir__}/datadog_api_client/v2", namespace: DatadogAPIClient::V2)
 loader.inflector = DatadogAPIClient::DatadogAPIClientInflector.new
 loader.setup