diff --git a/src/blenderbim/blenderbim/bim/module/system/ui.py b/src/blenderbim/blenderbim/bim/module/system/ui.py
index 6e20514bac..c70438292f 100644
--- a/src/blenderbim/blenderbim/bim/module/system/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/system/ui.py
@@ -24,8 +24,8 @@ from blenderbim.bim.module.system.data import SystemData, ObjectSystemData, Port
 
 
 FLOW_DIRECTION_TO_ICON = {
-    "SOURCE": "REMOVE",
-    "SINK": "ADD",
+    "SOURCE": "FORWARD",
+    "SINK": "BACK",
     "SOURCEANDSINK": "ARROW_LEFTRIGHT",
     "NOTDEFINED": "CHECKBOX_DEHLT",
 }