             configureWith(image: image, title: title, message: message)
         case .withButton(let errorImage, let title, let message, let buttonTitle,
                          let buttonConfig, let buttonAction):
+            buttonConfigCallback = buttonConfig
             configWithButton(image: errorImage, title: title, message: message,
                              buttonTitle: buttonTitle,
                              buttonConfig: buttonConfig,

    var actionButton = UIButton.autolayoutView()
    
    var buttonAction: (() -> Void)?
    
    public var currentState: SDStateTableViewState = .unknown {
        didSet {

        actionButton.layer.borderColor = buttonColor.cgColor
        actionButton.addTarget(self, action: #selector(self.retryButtonTapped(_:)), for: .touchUpInside)
        
        stackView.axis  = .vertical
        stackView.distribution  = .equalSpacing
        stackView.alignment = .center
