diff --git a/app/components/budgets/investments/map_component.html.erb b/app/components/budgets/investments/map_component.html.erb
index 2324476b8..d8961a740 100644
--- a/app/components/budgets/investments/map_component.html.erb
+++ b/app/components/budgets/investments/map_component.html.erb
@@ -1,3 +1,3 @@
-<div class="map">
+<div class="budget-investments-map">
   <%= render_map(map_location, investments_coordinates: coordinates) %>
 </div>
diff --git a/app/components/budgets/investments/map_component.rb b/app/components/budgets/investments/map_component.rb
index 7c88ac828..c843c38a4 100644
--- a/app/components/budgets/investments/map_component.rb
+++ b/app/components/budgets/investments/map_component.rb
@@ -1,5 +1,6 @@
 class Budgets::Investments::MapComponent < ApplicationComponent
   attr_reader :heading, :investments
+  delegate :render_map, to: :helpers
 
   def initialize(investments, heading:)
     @investments = investments
diff --git a/app/components/budgets/map_component.html.erb b/app/components/budgets/map_component.html.erb
index fa2284472..fb89cd1f0 100644
--- a/app/components/budgets/map_component.html.erb
+++ b/app/components/budgets/map_component.html.erb
@@ -1,4 +1,4 @@
-<div class="map inline">
+<div class="budgets-map">
   <h2><%= t("budgets.index.map") %></h2>
   <%= render_map(nil, investments_coordinates: coordinates) %>
 </div>
diff --git a/spec/components/budgets/map_component_spec.rb b/spec/components/budgets/map_component_spec.rb
index 7d9d3ffb8..60996af24 100644
--- a/spec/components/budgets/map_component_spec.rb
+++ b/spec/components/budgets/map_component_spec.rb
@@ -10,7 +10,7 @@ describe Budgets::MapComponent do
 
       render_inline Budgets::MapComponent.new(budget)
 
-      expect(page.first("div.map")).to have_content "located geographically"
+      expect(page.find(".budgets-map")).to have_content "located geographically"
     end
 
     it "is not rendered during the informing phase" do