diff --git a/components/webdriver_server/lib.rs b/components/webdriver_server/lib.rs
index 22776126a94..ecead687c1f 100644
--- a/components/webdriver_server/lib.rs
+++ b/components/webdriver_server/lib.rs
@@ -615,12 +615,13 @@ impl Handler {
             sleep_ms(interval)
         }
 
-        if img.is_none() {
-            return Err(WebDriverError::new(ErrorStatus::Timeout,
-                                           "Taking screenshot timed out"));
-        }
+        let mut img = match img {
+            Some(img) => img,
+            None => return Err(WebDriverError::new(ErrorStatus::Timeout,
+                                                   "Taking screenshot timed out")),
+        };
 
-        let img_vec = match png::to_vec(&mut img.unwrap()) {
+        let img_vec = match png::to_vec(&mut img) {
            Ok(x) => x,
            Err(_) => return Err(WebDriverError::new(ErrorStatus::UnknownError,
                                                     "Taking screenshot failed"))