diff --git a/index.js b/index.js
index d94efb6..71b84ef 100644
--- a/index.js
+++ b/index.js
@@ -108,6 +108,7 @@ module.exports.handler = async (event, context, callback) => {
       console.log(`Elements with selector: ${selectorToRemove} removed`);
     }
 
+    page.waitForTimeout(500);
     image = await element.screenshot({ type: 'png', encoding: 'base64' });
     console.log('Made screenshot');