diff --git a/core/assets/bundles/bundle.properties b/core/assets/bundles/bundle.properties
index 4ae1258e7d..b16bdbdcfa 100644
--- a/core/assets/bundles/bundle.properties
+++ b/core/assets/bundles/bundle.properties
@@ -401,6 +401,7 @@ waves.sort.reverse = Reverse Sort
 waves.sort.begin = Begin
 waves.sort.health = Health
 waves.sort.type = Type
+waves.units.hide = Hide All
 
 #these are intentionally in lower case
 wavemode.counts = counts
diff --git a/core/src/mindustry/editor/WaveGraph.java b/core/src/mindustry/editor/WaveGraph.java
index 76bbc82d2b..dc7dc6f112 100644
--- a/core/src/mindustry/editor/WaveGraph.java
+++ b/core/src/mindustry/editor/WaveGraph.java
@@ -181,6 +181,17 @@ public class WaveGraph extends Table{
 
         colors.clear();
         colors.left();
+        colors.button("@waves.units.hide", Styles.fullTogglet, () -> {
+            if(hidden.size == usedCopy.size){
+                hidden.clear();
+            }else{
+                hidden.addAll(usedCopy);
+            }
+
+            used.clear();
+            used.addAll(usedCopy);
+            for(UnitType o : hidden) used.remove(o);
+        }).update(b -> b.setChecked(hidden.size == usedCopy.size)).height(32f).width(130f);
         colors.pane(t -> {
             t.left();
             for(UnitType type : used){