Update tests to reflect latest pygments