diff --git a/flatland/utils/graphics_layer.py b/flatland/utils/graphics_layer.py
index 1b2ff7ea0c01240b4293e7386e8f0c358c4e2c96..2598a8893cf47ac9be526a7c8809f55ca2bb8ddf 100644
--- a/flatland/utils/graphics_layer.py
+++ b/flatland/utils/graphics_layer.py
@@ -66,7 +66,7 @@ class GraphicsLayer(object):
     def get_cmap(self, *args, **kwargs):
         return plt.get_cmap(*args, **kwargs)
 
-    def setRailAt(self, row, col, binTrans, target=None):
+    def setRailAt(self, row, col, binTrans, iTarget=None):
         """ Set the rail at cell (row, col) to have transitions binTrans.
             The target argument can contain the index of the agent to indicate
             that agent's target is at that cell, so that a station can be
diff --git a/flatland/utils/render_qt.py b/flatland/utils/render_qt.py
index ff8ebd12a966c76342dec8298fab2755b3734223..dc2217c3ee9041caf39c88d8898bf3fb0c7cb24e 100644
--- a/flatland/utils/render_qt.py
+++ b/flatland/utils/render_qt.py
@@ -157,7 +157,7 @@ class QTSVG(GraphicsLayer):
         self.lwAgents = []
         self.agents_prev = []
 
-    def setRailAt(self, row, col, binTrans, target=None):
+    def setRailAt(self, row, col, binTrans, iTarget=None):
         if binTrans in self.track.dSvg:
             sSVG = self.track.dSvg[binTrans].to_string()
             svgWidget = create_QtSvgWidget_from_svg_string(sSVG)