fixed wrong class names