diff --git a/display/epd4in0e_driver.py b/display/epd4in0e_driver.py
index cbec74f..652f1cc 100644
--- a/display/epd4in0e_driver.py
+++ b/display/epd4in0e_driver.py
@@ -22,7 +22,7 @@ class EPD4in0eDisplay(Display):
         self._epd = None
 
     def init(self) -> None:
-        from waveshare_epd import epd4in0e
+        from display.waveshare_epd import epd4in0e
 
         self._epd = epd4in0e.EPD()
         self._epd.init()