diff --git a/tests/test_generate_bytes_img_function_unit.py b/tests/test_generate_bytes_img_function_unit.py
index 215c3a0..61d8a21 100644
--- a/tests/test_generate_bytes_img_function_unit.py
+++ b/tests/test_generate_bytes_img_function_unit.py
@@ -44,6 +44,9 @@ async def querySelector(self, selector):
     async def screenshot(self, **kwargs):
         return 'screenshot !'
 
+    async def close(self):
+        pass
+
 class FakeBrowser:
     async def newPage(self):
         return FakePage()