Fixed wrong class name