diff --git a/.config/zsh/functions/prompt_tastytea_setup b/.config/zsh/functions/prompt_tastytea_setup
index 7fc84bb..74c6e08 100644
--- a/.config/zsh/functions/prompt_tastytea_setup
+++ b/.config/zsh/functions/prompt_tastytea_setup
@@ -150,6 +150,8 @@ function prompt_tastytea_setup() {
     PS2='%F{${prompt_tastytea_colours[prompt]}}%_>%f '
     PS3='%F{${prompt_tastytea_colours[prompt]}}?#%f '
     PS4='%F{${prompt_tastytea_colours[prompt]}}+%N:%i>%f '
+
+    prompt_cleanup 'prompt_tastytea_cleanup'
 }
 
 function prompt_tastytea_cleanup() {
@@ -159,7 +161,6 @@ function prompt_tastytea_cleanup() {
 }
 
 prompt_tastytea_setup "$@"
-prompt_cleanup 'prompt_tastytea_cleanup'
 
 # Local Variables:
 # mode: shell-script