add

alphagov · Aug 4, 2022 · a2f7c72 · a2f7c72
1 parent 4a03ddb
commit a2f7c72
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/govuk_app_config/govuk_prometheus_exporter.rb b/lib/govuk_app_config/govuk_prometheus_exporter.rb
@@ -10,6 +10,7 @@ def self.configure
       if defined?(Sidekiq)
         Sidekiq.configure_server do |config|
           require "prometheus_exporter/instrumentation"
+          require 'prometheus_exporter/client'
           config.server_middleware do |chain|
             chain.add PrometheusExporter::Instrumentation::Sidekiq
           end
@@ -29,7 +30,7 @@ def self.configure
       if defined?(Rails)
         Rails.application.middleware.unshift PrometheusExporter::Middleware
       else defined?(Sinatra)
-        Sinatra.application.middleware.unshift PrometheusExporter::Middleware
+        use PrometheusExporter::Middleware
       end
     end
   end