Fix, wrong class name