diff --git a/packages/three_js_helpers/lib/infinite_grid_helper.dart b/packages/three_js_helpers/lib/infinite_grid_helper.dart
index 1f7a3d8..1f61f00 100644
--- a/packages/three_js_helpers/lib/infinite_grid_helper.dart
+++ b/packages/three_js_helpers/lib/infinite_grid_helper.dart
@@ -38,7 +38,7 @@ class InfiniteGridHelper extends Mesh {
     uniform float uDistance;
 
     void main() {
-        vec3 pos = position.$axes * uDistance;
+        vec3 pos = position.${axes.name} * uDistance;
         pos.$planeAxes += cameraPosition.$planeAxes;
         
         worldPosition = pos;