         //
         if case SDStateTableViewState.withButton(_, _, _, _, _, _) = currentState {
             actionButton.isHidden = false
+            buttonConfigCallback(actionButton)
         } else {
             actionButton.isHidden = true
         }

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

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