Fixed a missing base class in the python interface