Fix STI class name