Small class inheritance problem correction