From ae1d63501e28ca34639abc637edc416840a13d72 Mon Sep 17 00:00:00 2001
From: Jonathan Slenders <jonathan@slenders.be>
Date: Wed, 25 Sep 2024 12:08:29 +0200
Subject: [PATCH] Fixup for previous cursorshapes commit (#1900). (#1920)

---
 src/prompt_toolkit/cursor_shapes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/prompt_toolkit/cursor_shapes.py b/src/prompt_toolkit/cursor_shapes.py
index cbd8e593f..01d10926a 100644
--- a/src/prompt_toolkit/cursor_shapes.py
+++ b/src/prompt_toolkit/cursor_shapes.py
@@ -78,7 +78,7 @@ def get_cursor_shape(self, application: Application[Any]) -> CursorShape:
                 InputMode.INSERT_MULTIPLE,
             }:
                 return CursorShape.BEAM
-            if application.vi_state.input_mode == {
+            if application.vi_state.input_mode in {
                 InputMode.REPLACE,
                 InputMode.REPLACE_SINGLE,
             }: