Small refactor to put voxel-specific key func back in its package