diff --git a/litedram/phy/gw2ddrphy.py b/litedram/phy/gw2ddrphy.py
index 9bb474a..4cdc897 100644
--- a/litedram/phy/gw2ddrphy.py
+++ b/litedram/phy/gw2ddrphy.py
@@ -19,7 +19,7 @@ from migen import *
 from migen.fhdl.specials import Tristate
 from migen.genlib.cdc import MultiReg
 
-from migen.genlib.misc import timeline, WaitTimer
+from litex.gen.genlib.misc import timeline, WaitTimer
 
 from litex.soc.interconnect.csr import *