diff --git a/SVWebViewController/SVModalWebViewController.m b/SVWebViewController/SVModalWebViewController.m
index 83f2337..d7f1ac5 100644
--- a/SVWebViewController/SVModalWebViewController.m
+++ b/SVWebViewController/SVModalWebViewController.m
@@ -60,7 +60,7 @@ - (void)viewWillAppear:(BOOL)animated {
 #pragma mark - Delegate
 
 - (void)setWebViewDelegate:(id<UIWebViewDelegate>)webViewDelegate {
-    self.webViewController.delegate = webViewDelegate;
+    self.webViewController.delegate = (id<SVWebViewControllerDelegate>)webViewDelegate;
 }
 
 - (id<UIWebViewDelegate>)webViewDelegate {
diff --git a/SVWebViewController/SVWebViewController.h b/SVWebViewController/SVWebViewController.h
index b41ce2c..c15c9aa 100644
--- a/SVWebViewController/SVWebViewController.h
+++ b/SVWebViewController/SVWebViewController.h
@@ -6,12 +6,16 @@
 //
 //  https://github.com/samvermette/SVWebViewController
 
+@protocol SVWebViewControllerDelegate <UIWebViewDelegate>
+-(void) didPressDoneButton;
+@end
+
 @interface SVWebViewController : UIViewController
 
 - (instancetype)initWithAddress:(NSString*)urlString;
 - (instancetype)initWithURL:(NSURL*)URL;
 - (instancetype)initWithURLRequest:(NSURLRequest *)request;
 
-@property (nonatomic, weak) id<UIWebViewDelegate> delegate;
+@property (nonatomic, weak) id<SVWebViewControllerDelegate> delegate;
 
 @end
diff --git a/SVWebViewController/SVWebViewController.m b/SVWebViewController/SVWebViewController.m
index b9b2aea..16b9ecf 100644
--- a/SVWebViewController/SVWebViewController.m
+++ b/SVWebViewController/SVWebViewController.m
@@ -304,8 +304,12 @@ - (void)actionButtonTapped:(id)sender {
     }
 }
 
-- (void)doneButtonTapped:(id)sùender {
-    [self dismissViewControllerAnimated:YES completion:NULL];
+- (void)doneButtonTapped:(id)sender {
+    [self dismissViewControllerAnimated:YES completion:^{
+      if ([self.delegate respondsToSelector:@selector(didPressDoneButton)]) {
+        [self.delegate didPressDoneButton];
+      }
+    }];
 }
 
 @end