diff --git a/lib/model/HGPIFuNet.py b/lib/model/HGPIFuNet.py
index 477171534..2a02c705c 100755
--- a/lib/model/HGPIFuNet.py
+++ b/lib/model/HGPIFuNet.py
@@ -72,7 +72,7 @@ def query(self, points, calibs, transforms=None, labels=None):
         store all intermediate features.
         query() function may behave differently during training/testing.
         :param points: [B, 3, N] world space coordinates of points
-        :param calibs: [B, 3, 4] calibration matrices for each image
+        :param calibs: [B, 4, 4] calibration matrices for each image
         :param transforms: Optional [B, 2, 3] image space coordinate transforms
         :param labels: Optional [B, Res, N] gt labeling
         :return: [B, Res, N] predictions for each point