fix issue caution by wrong class name