diff --git a/gears.scad b/gears.scad
index a32309c..51e2e51 100644
--- a/gears.scad
+++ b/gears.scad
@@ -1784,7 +1784,7 @@ module worm_gear(
         difference() {
             vnf_polyhedron(vnf, convexity=teeth/2);
             if (shaft_diam > 0) {
-                cylinder(d=shaft_diam, l=worm_diam, center=true);
+                cylinder(h=2*thickness+1, r=shaft_diam/2, center=true, $fn=max(12,segs(shaft_diam/2)));
             }
         }
         children();