diff --git a/wave_viewer.gd b/wave_viewer.gd
index df90470..aeadb96 100644
--- a/wave_viewer.gd
+++ b/wave_viewer.gd
@@ -11,9 +11,10 @@ signal closed()
 
 
 func _unhandled_input(event: InputEvent) -> void:
-	if event.is_action_pressed("Pause"):
+	if event.is_action_pressed("Pause") or event.is_action_pressed("Show Wave Preview"):
 		queue_free()
 		closed.emit()
+		get_viewport().set_input_as_handled()
 
 
 func set_waves(waves: Array[WaveConfig], starting_wave_number: int) -> void: