From bb1258f2b2376a180181602ffc65ee29a6716f47 Mon Sep 17 00:00:00 2001
From: kikito <kikito@gmail.com>
Date: Thu, 22 Dec 2016 20:27:46 +0100
Subject: [PATCH] Fixes view

---
 .../budgets/investments/_header.html.erb      | 24 ++++++++-----------
 1 file changed, 10 insertions(+), 14 deletions(-)

diff --git a/app/views/budgets/investments/_header.html.erb b/app/views/budgets/investments/_header.html.erb
index 468d1b3c6..610d55186 100644
--- a/app/views/budgets/investments/_header.html.erb
+++ b/app/views/budgets/investments/_header.html.erb
@@ -1,4 +1,4 @@
-<% if @filter_heading_name.present? %>
+<% if @heading.present? %>
   <section class="no-margin-top margin-bottom">
     <div class="expanded jumbo-budget budget-heading padding">
 
@@ -27,20 +27,16 @@
                             budget_ballot_path(@budget) %>
               <% end %>
             </div>
-              <% if false
-                    #@heading.present? &&
-                    #@ballot.heading.present? &&
-                    #@ballot.heading != @heading
-              %>
-              <br>
-              <p class="callout warning inline-block">
-                <%= t("budget.investments.header.different_heading_active") %>
-                <%= link_to @ballot.heading.name, budget_investments_path(budget_id: budget.id, heading_id: @ballot.heading_id) %>
-              </p>
+              <% if @ballot.valid_heading?(@heading) %>
+                <div id="progress_bar" class="no-margin-top">
+                  <%= render 'budgets/ballot/progress_bar' %>
+                </div>
               <% else %>
-              <div id="progress_bar" class="no-margin-top">
-                <%= render 'budgets/ballot/progress_bar' %>
-              </div>
+                <br>
+                <p class="callout warning inline-block">
+                  <%= t("budget.investments.header.different_heading_active") %>
+                  <%= link_to @ballot.heading.name, budget_investments_path(budget_id: budget.id, heading_id: @ballot.heading_id) %>
+                </p>
               <% end %>
           </div>
         </div>