diff --git a/Classes/IDMPhotoBrowser.m b/Classes/IDMPhotoBrowser.m
index 7a4e4a32..125785fa 100644
--- a/Classes/IDMPhotoBrowser.m
+++ b/Classes/IDMPhotoBrowser.m
@@ -615,7 +615,7 @@ - (void)viewDidLoad {
         _doneButton.layer.cornerRadius = 3.0f;
         _doneButton.layer.borderColor = [UIColor colorWithWhite:0.9 alpha:0.9].CGColor;
         _doneButton.layer.borderWidth = 1.0f;
-        _doneButtonSize = _doneButtonImage.size;
+        _doneButtonSize = _doneButton.frame.size;
     }
     else {
         [_doneButton setImage:_doneButtonImage forState:UIControlStateNormal];