diff --git a/minigrid/envs/wfc/graphtransforms.py b/minigrid/envs/wfc/graphtransforms.py
index 8afb64313..c2ea41297 100644
--- a/minigrid/envs/wfc/graphtransforms.py
+++ b/minigrid/envs/wfc/graphtransforms.py
@@ -15,7 +15,6 @@
 class EdgeDescriptor:
     between: tuple[str, str] | tuple[str]
     structure: str | None = None
-    weight = None
 
 
 # This is maybe general enough to be in utils
diff --git a/minigrid/envs/wfc/wfcenv.py b/minigrid/envs/wfc/wfcenv.py
index 281391b15..c021a82de 100644
--- a/minigrid/envs/wfc/wfcenv.py
+++ b/minigrid/envs/wfc/wfcenv.py
@@ -102,7 +102,7 @@ def __init__(
         if size < 3:
             raise ValueError(f"Grid size must be at least 3 (currently {size})")
         self.size = size
-        self.max_attempts = 100
+        self.max_attempts = 1000
 
         if max_steps is None:
             max_steps = self.size * 20
diff --git a/tests/test_wfc/conftest.py b/tests/test_wfc/conftest.py
index d5a9be0bd..1f88603da 100644
--- a/tests/test_wfc/conftest.py
+++ b/tests/test_wfc/conftest.py
@@ -22,7 +22,7 @@ def img_redmaze(resources: Resources) -> array:
         import imageio  # type: ignore
 
         pattern = resources.get_pattern("RedMaze.png")
-        img = imageio.imread(pattern)
+        img = imageio.v2.imread(pattern)
     except ImportError:
         b = [0, 0, 0]
         w = [255, 255, 255]