diff --git a/api/controller/image_unittest.py b/api/controller/image_unittest.py
index 8b2c93b..5d45a17 100644
--- a/api/controller/image_unittest.py
+++ b/api/controller/image_unittest.py
@@ -385,7 +385,7 @@
     def _CreateMockBuildResult(
         self, image_types: List[int]
     ) -> Optional[image_service.BuildResult]:
-        """Helper to create Mock build_image results.
+        """Helper to create Mock `cros build-image` results.
 
         Args:
             image_types: A list of image types for which the mock BuildResult