diff --git a/functions/_sponge_on_prompt.fish b/functions/_sponge_on_prompt.fish
index b12645f..03e989a 100644
--- a/functions/_sponge_on_prompt.fish
+++ b/functions/_sponge_on_prompt.fish
@@ -1,3 +1,5 @@
 function _sponge_on_prompt --on-event fish_prompt
-  _sponge_remove_from_history
+  if test $sponge_purge_only_on_exit = false
+    _sponge_remove_from_history
+  end
 end