diff --git a/masque/shapes/path.py b/masque/shapes/path.py
index 706e2f4..a7b0ac6 100644
--- a/masque/shapes/path.py
+++ b/masque/shapes/path.py
@@ -486,9 +486,7 @@ class Path(Shape):
         return self
 
     def _calculate_cap_extensions(self) -> NDArray[numpy.float64]:
-        if self.cap == PathCap.Square:
-            extensions = numpy.full(2, self.width / 2)
-        elif self.cap == PathCap.Circle:
+        if self.cap in (PathCap.Square, PathCap.Circle):
             extensions = numpy.full(2, self.width / 2)
         elif self.cap == PathCap.SquareCustom:
             assert isinstance(self.cap_extensions, numpy.ndarray)