diff --git a/Source/LightboxImage.swift b/Source/LightboxImage.swift
index 84aac95..9885f60 100644
--- a/Source/LightboxImage.swift
+++ b/Source/LightboxImage.swift
@@ -38,7 +38,10 @@ open class LightboxImage {
       imageView.image = image
       completion?(image)
     } else if let imageURL = imageURL {
-      LightboxConfig.loadImage(imageView, imageURL, completion)
+      LightboxConfig.loadImage(imageView, imageURL) { [weak self] downloadedImage in
+        self?.image = downloadedImage
+        completion?(downloadedImage)
+      }
     } else if let imageClosure = imageClosure {
       let img = imageClosure()
       imageView.image = img