diff --git a/codes/models/lucidrains/performer/__init__.py b/codes/models/lucidrains/performer/__init__.py
index 0dfe9687..8b137891 100644
--- a/codes/models/lucidrains/performer/__init__.py
+++ b/codes/models/lucidrains/performer/__init__.py
@@ -1,3 +1 @@
-from performer_pytorch import PerformerLM, Performer, FastAttention, SelfAttention, CrossAttention, ProjectionUpdater
-from autoregressive_wrapper import AutoregressiveWrapper
-from performer_enc_dec import PerformerEncDec
+
diff --git a/codes/models/lucidrains/performer/performer_pytorch.py b/codes/models/lucidrains/performer/performer_pytorch.py
index 836ed03f..2d618f11 100644
--- a/codes/models/lucidrains/performer/performer_pytorch.py
+++ b/codes/models/lucidrains/performer/performer_pytorch.py
@@ -10,7 +10,7 @@ from contextlib import contextmanager
 
 from local_attention import LocalAttention
 from axial_positional_embedding import AxialPositionalEmbedding
-from reversible import ReversibleSequence, SequentialSequence
+from models.lucidrains.performer.reversible import ReversibleSequence, SequentialSequence
 
 from distutils.version import LooseVersion