iOS UIView скругление верхних углов работает только для верхнего левого

Question

Я хочу закруглить верхние углы UIView. Эмулятор iOS версии 12.0

Мое расширение

    extension UIView
{
    func round(corners: UIRectCorner, radius: CGFloat) {
        let path = UIBezierPath(roundedRect: bounds, byRoundingCorners: corners, cornerRadii: CGSize(width: radius, height: radius))
        let mask = CAShapeLayer()
        mask.path = path.cgPath
        self.layer.mask = mask
    }
}

Как я это использую

test.round(corners: [.topRight, .topLeft], radius: 20)

Результат