diff --git a/qr_nodes.py b/qr_nodes.py
index ac24431..9061fd3 100644
--- a/qr_nodes.py
+++ b/qr_nodes.py
@@ -7,7 +7,7 @@ from qrcode.image.styles.moduledrawers import (GappedSquareModuleDrawer,
                                                HorizontalBarsDrawer)
 from qrcode.image.styledpil import StyledPilImage
 from qrcode.image.styles.colormasks import SolidFillColorMask
-from qrcode.compat.pil import Image
+from PIL import Image
 import torch
 import torch.nn.functional as F
 
diff --git a/requirements.txt b/requirements.txt
index 4622fb7..3559057 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1,2 @@
-qrcode >= 7.1
\ No newline at end of file
+qrcode >= 7.1
+pillow >= 9.0