diff --git a/include/polyscope/volume_grid.h b/include/polyscope/volume_grid.h
index dcc504e7..98addb93 100644
--- a/include/polyscope/volume_grid.h
+++ b/include/polyscope/volume_grid.h
@@ -80,8 +80,6 @@ class VolumeGrid : public QuantityStructure<VolumeGrid> {
   template <class Func>
   VolumeGridCellScalarQuantity* addCellScalarQuantityFromBatchCallable(std::string name, Func&& func, DataType dataType_ = DataType::STANDARD);
 
-
-  //template <class T> VolumeGridScalarIsosurface* addGridIsosurfaceQuantity(std::string name, double isoLevel, const T& values);
   
   // Rendering helpers used by quantities
   // void populateGeometry();
@@ -195,7 +193,7 @@ class VolumeGrid : public QuantityStructure<VolumeGrid> {
 
 
 VolumeGrid* registerVolumeGrid(std::string name, glm::uvec3 gridNodeDim, glm::vec3 boundMin, glm::vec3 boundMax);
-VolumeGrid* registerVolumeGrid(std::string name, uint64_t gridNodeDim, glm::vec3 boundMin, glm::vec3 boundMax);
+VolumeGrid* registerVolumeGrid(std::string name, uint64_t gridNodeAxesDim, glm::vec3 boundMin, glm::vec3 boundMax);
 
 // Shorthand to get a point cloud from polyscope
 inline VolumeGrid* getVolumeGrid(std::string name = "");
diff --git a/src/volume_grid.cpp b/src/volume_grid.cpp
index 43edd131..84ea44f6 100644
--- a/src/volume_grid.cpp
+++ b/src/volume_grid.cpp
@@ -539,7 +539,7 @@ VolumeGrid* registerVolumeGrid(std::string name, glm::uvec3 gridNodeDim, glm::ve
   return s;
 }
 
-VolumeGrid* registerVolumeGrid(std::string name, size_t gridNodeDim, glm::vec3 boundMin, glm::vec3 boundMax) {
+VolumeGrid* registerVolumeGrid(std::string name, uint64_t gridNodeDim, glm::vec3 boundMin, glm::vec3 boundMax) {
   return registerVolumeGrid(name, {gridNodeDim, gridNodeDim, gridNodeDim}, boundMin, boundMax);
 }