diff --git a/kitty/boss.py b/kitty/boss.py index 8d2d52ae7..6d8c3f726 100644 --- a/kitty/boss.py +++ b/kitty/boss.py @@ -58,9 +58,9 @@ def data_for_at(w, arg, add_wrap_markers=False): if arg == '@selection': return w.text_for_selection() - if arg == '@ansi' or arg == '@ansi_screen_scrollback': + if arg in ('@ansi', '@ansi_screen_scrollback'): return as_text(as_ansi=True, add_history=True) - if arg == '@text' or arg == '@screen_scrollback': + if arg in ('@text', '@screen_scrollback'): return as_text(add_history=True) if arg == '@screen': return as_text()