Fixed incorrect base class