Swift - Удлинение с закругленными краями только на 2 ребра на 4"устройствах

Question

Я работаю над приложением с расширением Round Edges. Работает нормально, кроме как на 4"устройствах (см. Изображение). Что я могу сделать?

Спасибо!

Вот мой код

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