Add compatibility with Python visualizers