diff --git a/litejesd204b/core.py b/litejesd204b/core.py
index 5697c97..8e2e71c 100644
--- a/litejesd204b/core.py
+++ b/litejesd204b/core.py
@@ -6,11 +6,12 @@
 # SPDX-License-Identifier: BSD-2-Clause
 
 from migen import *
-from migen.genlib.cdc import MultiReg, ElasticBuffer
+from migen.genlib.cdc import MultiReg
 from migen.genlib.fifo import SyncFIFO
 
 from litex.gen import *
 from litex.gen.genlib.misc import WaitTimer
+from litex.gen.genlib.cdc import ElasticBuffer
 
 from litex.build.io import DifferentialInput, DifferentialOutput