diff --git a/shell-integration/zsh/kitty-integration b/shell-integration/zsh/kitty-integration
index 5195f1791..59c8137e0 100644
--- a/shell-integration/zsh/kitty-integration
+++ b/shell-integration/zsh/kitty-integration
@@ -181,11 +181,7 @@ _ksi_deferred_init() {
                     # - False positive (with prompt_percent): PS1="%(?.$mark1.)"
                     # - False negative (with prompt_subst):   PS1='$mark1'
                     [[ $PS1 == *$mark1* ]] || PS1=${mark1}${PS1}
-                    # The following line is commented out because currently kitty doesn't
-                    # use B prompt marking, so there is no need to pay performance penalty
-                    # for it.
-                    #
-                    #   [[ $PS2 == *$mark2* ]] || PS2=${mark2}${PS2}
+                    [[ $PS2 == *$mark2* ]] || PS2=${mark2}${PS2}
                     (( _ksi_state = 2 ))
                 else
                     # If our precmd hook is not the last, we cannot rely on prompt
@@ -224,12 +220,7 @@ _ksi_deferred_init() {
             # top. We cannot force prompt_subst on the user though, so we would
             # still need this code for the no_prompt_subst case.
             PS1=${PS1//$'%{\e]133;A\a%}'}
-
-            # The following line is commented out because currently kitty doesn't
-            # use B prompt marking, so there is no need to pay performance penalty
-            # for it.
-            #
-            #   PS2=${PS2//$'%{\e]133;A;k=s\a%}'}
+            PS2=${PS2//$'%{\e]133;A;k=s\a%}'}
 
             # This will work incorrectly in the presence of a preexec hook that
             # prints. For example, if MichaelAquilina/zsh-you-should-use installs