diff --git a/SketchWidget.py b/SketchWidget.py
index 192d091..049d668 100644
--- a/SketchWidget.py
+++ b/SketchWidget.py
@@ -727,20 +727,18 @@ class DimMode(Mode):
 
 class XDimMode(DimMode):
     def get_constraint(self):
-        if (self.ptrefs[0].shape.getPt(self.ptrefs[0].ptNum)[0] <
-                self.ptrefs[1].shape.getPt(self.ptrefs[1].ptNum)[0]):
+        if (self.ptrefs[0].getPt()[0] < self.ptrefs[1].getPt()[0]):
             p1, p2 = self.ptrefs
         else:
             p2, p1 = self.ptrefs
-        dist = (p2.shape.getPt(p2.ptNum)[0] - p1.shape.getPt(p1.ptNum)[0])
+        dist = p2.getPt()[0] - p1.getPt()[0]
         return XDistance(p1.shape, p1.ptNum, p2.shape, p2.ptNum, dist)
 
 class YDimMode(DimMode):
     def get_constraint(self):
-        if (self.ptrefs[0].shape.getPt(self.ptrefs[0].ptNum)[1] <
-                self.ptrefs[1].shape.getPt(self.ptrefs[1].ptNum)[1]):
+        if (self.ptrefs[0].getPt()[1] < self.ptrefs[1].getPt()[1]):
             p1, p2 = self.ptrefs
         else:
             p2, p1 = self.ptrefs
-        dist = (p2.shape.getPt(p2.ptNum)[1] - p1.shape.getPt(p1.ptNum)[1])
+        dist = p2.getPt()[1] - p1.getPt()[1]
         return YDistance(p1.shape, p1.ptNum, p2.shape, p2.ptNum, dist)