diff --git a/q_transformer/robotic_transformer.py b/q_transformer/robotic_transformer.py
index 0dc684a..70dcd62 100644
--- a/q_transformer/robotic_transformer.py
+++ b/q_transformer/robotic_transformer.py
@@ -811,5 +811,4 @@ def forward(
         if self.num_actions == 1:
             q_values = rearrange(q_values, '... 1 b -> ... b')
 
-        exit()
         return q_values
diff --git a/setup.py b/setup.py
index 8587f44..1172236 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'q-transformer',
   packages = find_packages(exclude=[]),
-  version = '0.0.14',
+  version = '0.0.15',
   license='MIT',
   description = 'Q-Transformer',
   author = 'Phil Wang',