diff --git a/trap/animation_renderer.py b/trap/animation_renderer.py
index 58eddd4..c968d03 100644
--- a/trap/animation_renderer.py
+++ b/trap/animation_renderer.py
@@ -170,9 +170,6 @@ class AnimationRenderer:
                 
         
 
-        
-        self.init_shapes()
-
         self.init_labels()
 
     
@@ -200,52 +197,6 @@ class AnimationRenderer:
         )
         # return process
 
-        
-
-    def init_shapes(self):
-        '''
-        Due to error when running headless, we need to configure options before extending the shapes class
-        '''
-        class GradientLine(shapes.Line):
-            def __init__(self, x, y, x2, y2, width=1, color1=[255,255,255], color2=[255,255,255], batch=None, group=None):
-                # print('colors!', colors)
-                # assert len(colors) == 6
-                
-                r, g, b, *a = color1
-                self._rgba1 = (r, g, b, a[0] if a else 255)
-                r, g, b, *a = color2
-                self._rgba2 = (r, g, b, a[0] if a else 255)
-                
-                # print('rgba', self._rgba)
-
-                super().__init__(x, y, x2, y2, width, color1, batch=None, group=None)
-                # <pyglet.graphics.vertexdomain.VertexList
-                # pyglet.graphics.vertexdomain
-                # print(self._vertex_list)
-            
-            def _create_vertex_list(self):
-                '''
-                copy of super()._create_vertex_list but with additional colors'''
-                self._vertex_list = self._group.program.vertex_list(
-                    6, self._draw_mode, self._batch, self._group,
-                    position=('f', self._get_vertices()),
-                    colors=('Bn', self._rgba1+ self._rgba2 + self._rgba2 + self._rgba1 + self._rgba2 +self._rgba1 ),
-                    translation=('f', (self._x, self._y) * self._num_verts))
-            
-            def _update_colors(self):
-                self._vertex_list.colors[:] = self._rgba1+ self._rgba2 + self._rgba2 + self._rgba1 + self._rgba2 +self._rgba1
-            
-            def color1(self, color):
-                r, g, b, *a = color
-                self._rgba1 = (r, g, b, a[0] if a else 255)
-                self._update_colors()
-            
-            def color2(self, color):
-                r, g, b, *a = color
-                self._rgba2 = (r, g, b, a[0] if a else 255)
-                self._update_colors()
-                
-        self.gradientLine = GradientLine
 
     def init_labels(self):
         base_color = COLOR_PRIMARY
diff --git a/trap/preview_renderer.py b/trap/preview_renderer.py
index e721dfa..97dc974 100644
--- a/trap/preview_renderer.py
+++ b/trap/preview_renderer.py
@@ -205,7 +205,9 @@ class DrawnTrack:
                     if draw_dot:
                         line = pyglet.shapes.Arc(x2, y2, 10, thickness=2, color=color, batch=self.renderer.batch_anim)
                     else:
-                        line = self.renderer.gradientLine(x, y, x2, y2, 3, color, color, batch=self.renderer.batch_anim)
+                        # line = self.renderer.gradientLine(x, y, x2, y2, 3, color, color, batch=self.renderer.batch_anim)
+                        line = pyglet.shapes.Line(x, y, x2, y2, 3, color, batch=self.renderer.batch_anim)
+                        # line = self.renderer.gradientLine(x, y, x2, y2, 3, color, color, batch=self.renderer.batch_anim)
                     line.opacity = 20 if not for_laser else 255
                     self.shapes.append(line)