diff --git a/autoload/vimtex/syntax/p/sagetex.vim b/autoload/vimtex/syntax/p/sagetex.vim index a413f3d659..8cf9e1b3e8 100644 --- a/autoload/vimtex/syntax/p/sagetex.vim +++ b/autoload/vimtex/syntax/p/sagetex.vim @@ -18,11 +18,14 @@ function! vimtex#syntax#p#sagetex#load(cfg) abort " {{{1 \}) call vimtex#syntax#core#new_opt('texSagetexOpt', {'next': 'texSagetexArg'}) - call vimtex#syntax#core#new_env({ - \ 'name': 'sagesilent', - \ 'region': 'texSagetexZone', - \ 'contains': '@vimtex_nested_python' - \}) + for sage_env in ['sageblock', 'sagesilent', 'sageverbatim', 'sageexample', 'sagecommandline'] + let config = { + \ 'name': sage_env, + \ 'region': 'texSagetexZone', + \ 'contains': '@vimtex_nested_python' + \} + call vimtex#syntax#core#new_env(config) + endfor highlight def link texCmdSagetex texCmd endfunction