Fixing inherited class name