--- a/eric6/Graphics/AssociationItem.py	Tue Apr 27 17:42:00 2021 +0200
+++ b/eric6/Graphics/AssociationItem.py	Wed Apr 28 19:42:28 2021 +0200
@@ -10,16 +10,17 @@
 from PyQt5.QtCore import QPointF, QRectF, QLineF
 from PyQt5.QtWidgets import QGraphicsItem
 
-from E5Graphics.E5ArrowItem import E5ArrowItem, NormalArrow, WideArrow
+from E5Graphics.E5ArrowItem import E5ArrowItem, E5ArrowType
 
 import Utilities
 
 
+# TODO: convert to Enum
 Normal = 0
 Generalisation = 1
 Imports = 2
 
-
+# TODO: convert to Enum
 NoRegion = 0
 West = 1
 North = 2
@@ -61,10 +62,10 @@
         @type QGraphicsItem
         """
         if assocType in (Normal, Imports):
-            arrowType = NormalArrow
+            arrowType = E5ArrowType.NORMAL
             arrowFilled = True
         elif assocType == Generalisation:
-            arrowType = WideArrow
+            arrowType = E5ArrowType.WIDE
             arrowFilled = False
         
         E5ArrowItem.__init__(self, QPointF(0, 0), QPointF(100, 100),