lib/skylight/probes/sinatra_add_middleware.rb in skylight-5.1.1 vs lib/skylight/probes/sinatra_add_middleware.rb in skylight-5.2.0.beta

@@ -1,10 +1,10 @@
 module Skylight
   module Probes
     module Sinatra
       module Instrumentation
-        def build(*)
-          use Skylight::Middleware
+        def setup_default_middleware(builder)
+          builder.use Skylight::Middleware
           super
         end
       end
 
       class AddMiddlewareProbe