diff --git a/q_transformer/__init__.py b/q_transformer/__init__.py
index ca7434f..867fb32 100644
--- a/q_transformer/__init__.py
+++ b/q_transformer/__init__.py
@@ -1,8 +1,8 @@
-from q_transformer.robotic_transformer import (
+from q_transformer.q_robotic_transformer import (
     QRoboticTransformer,
     MaxViT
 )
 
-from q_transformer.q_transformer import (
+from q_transformer.q_learner import (
     QLearner
 )
diff --git a/q_transformer/q_transformer.py b/q_transformer/q_learner.py
similarity index 99%
rename from q_transformer/q_transformer.py
rename to q_transformer/q_learner.py
index 15e15ff..3e900f0 100644
--- a/q_transformer/q_transformer.py
+++ b/q_transformer/q_learner.py
@@ -17,7 +17,7 @@
 from beartype import beartype
 from beartype.typing import Optional, Union, List, Tuple
 
-from q_transformer.robotic_transformer import QRoboticTransformer
+from q_transformer.q_robotic_transformer import QRoboticTransformer
 
 from q_transformer.optimizer import get_adam_optimizer
 
diff --git a/q_transformer/robotic_transformer.py b/q_transformer/q_robotic_transformer.py
similarity index 100%
rename from q_transformer/robotic_transformer.py
rename to q_transformer/q_robotic_transformer.py
diff --git a/setup.py b/setup.py
index 8803cfd..1500cb0 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'q-transformer',
   packages = find_packages(exclude=[]),
-  version = '0.0.23',
+  version = '0.0.24',
   license='MIT',
   description = 'Q-Transformer',
   author = 'Phil Wang',