Fixup for previous cursorshapes commit (prompt-toolkit#1900). (prompt-toolkit#1920)

Author: jonathanslenders

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,
             }: