Fix class capitalization