module 'tvtk.tvtk_classes.arrow_source' has no attribute 'ArrowSource' in off_screen_engine

This is the error:
```
Exception has occurred: AttributeError
module 'tvtk.tvtk_classes.arrow_source' has no attribute 'ArrowSource'
File "/path/site-packages/tvtk_classes/tvtk_helper.py", line 48, in get_class
File "/path/site-packages/tvtk_classes/tvtk_helper.py", line 809, in
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/tvtk/pyface/light_manager.py", line 51, in __init__
arrow = tvtk.ArrowSource()
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/tvtk/pyface/light_manager.py", line 167, in __init__
self.glyph = LightGlyph()
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/tvtk/pyface/light_manager.py", line 354, in __init__
light = CameraLight(self.renwin)
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/tvtk/pyface/tvtk_scene.py", line 845, in _create_control
self.light_manager = light_manager.LightManager(self)
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/tvtk/pyface/tvtk_scene.py", line 222, in __init__
self.control = self._create_control(parent)
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/tvtk/pyface/tvtk_scene.py", line 986, in __init__
self.scene = TVTKScene(**traits)
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/mayavi/core/off_screen_engine.py", line 16, in off_screen_viewer_factory
win = TVTKWindow(off_screen_rendering=True)
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/mayavi/core/engine.py", line 452, in new_scene
viewer = self.scene_factory(**factory_kwargs)
File "/path/site-packages/apptools/scripting/recordable.py", line 45, in _wrapper
result = func(*args, **kw)
File "/path/site-packages/mayavi-4.7.2.dev0-py3.7-linux-x86_64.egg/mayavi/tools/figure.py", line 69, in figure
engine.new_scene(name=name, size=size)
File "/home/t-dvginz/repos/utils_repos/CPD/visualization/mesh_visualization/mesh_visualizer.py", line 66, in visualize_mesh_pair
fig = mlab.figure(size=(600, 600), bgcolor=(0, 0, 0))
File "/path2/shape_correspondence_pl_template.py", line 199, in visualize_pair_corr
save_path=image_path)
File "/path2/shape_correspondence_pl_template.py", line 165, in validation_step
self.visualize_pair_corr(batch, P)
```
Ubuntu headless system, latest mayavi(from source) version.
This is how I import mayavi
```
from xvfbwrapper import Xvfb
vdisplay = Xvfb(width=1920, height=1080)
vdisplay.start()
from mayavi import mlab
mlab.options.offscreen = True
```