将FaveButton添加到自定义单元格

Question

我在自定义表格单元格中添加了收藏夹按钮。但是，当我单击几次时，它消失并弹出。 https://github.com/xhamr/fave-button

我只写了代码。

let faveButton = FaveButton(
        frame: CGRect(x:200, y:200, width: 44, height: 44),
        faveIconNormal: UIImage(named: "heart")
    )


func faveButton(_ faveButton: FaveButton, didSelected selected: Bool) {
    if faveButton.isSelected == true {
        faveButton.selectedColor = UIColor.red
    }
    else {
        faveButton.selectedColor = UIColor.clear
    }
}

初始颜色清晰，单击时应填充为红色。