diff --git a/examples/demo.py b/examples/demo.py
index dae2118c9a80680fbf89b38481c988ec7d404ed7..86e986f6fd0587b3d5781175c6c59565db3d4c71 100644
--- a/examples/demo.py
+++ b/examples/demo.py
@@ -127,7 +127,7 @@ class Demo:
         self.action_size = 4
 
     def create_renderer(self):
-        self.renderer = RenderTool(self.env, gl="QTSVG")
+        self.renderer = RenderTool(self.env, gl="PILSVG")
         handle = self.env.get_agent_handles()
         return handle
 
diff --git a/flatland/utils/graphics_pil.py b/flatland/utils/graphics_pil.py
index fe5e74b541b3d958df7058acc5b127c45d1a8ebd..63a4be28fa8f5fc1bc1f19e29d14ae74e766f587 100644
--- a/flatland/utils/graphics_pil.py
+++ b/flatland/utils/graphics_pil.py
@@ -351,7 +351,7 @@ class PILSVG(PILGL):
                     rgbaZug3[xy_color_mask, 0:3] = tnNewColor
                     self.dPilZug[(iDirIn2, iDirOut2, iColor)] = Image.fromarray(rgbaZug3)
 
-    def setAgentAt(self, iAgent, row, col, iDirIn, iDirOut):
+    def setAgentAt(self, iAgent, row, col, iDirIn, iDirOut, color=None):
         delta_dir = (iDirOut - iDirIn) % 4
         iColor = iAgent % self.nAgentColors
         # when flipping direction at a dead end, use the "iDirOut" direction.