diff --git a/scripts/plot_summary.py b/scripts/plot_summary.py
index bc852e70..9b3a81e1 100644
--- a/scripts/plot_summary.py
+++ b/scripts/plot_summary.py
@@ -177,7 +177,7 @@ def plot_balances():
     balances_df = pd.read_csv(snakemake.input.balances,index_col=list(range(3)),header=list(range(n_header)))
 
     balances = {i.replace(" ","_") : [i] for i in balances_df.index.levels[0]}
-    balances["energy"] = balances_df.index.levels[0].symmetric_difference(co2_carriers)
+    balances["energy"] = [i for i in balances_df.index.levels[0] if i not in co2_carriers]
 
     for k,v in balances.items():
 
@@ -401,4 +401,4 @@ if __name__ == "__main__":
         opts=sector_opts.split('-')
         for o in opts:
             if "cb" in o:
-                plot_carbon_budget_distribution()
\ No newline at end of file
+                plot_carbon_budget_distribution()