Set correct base class