Fixing base class reference