diff --git a/pyinfinitensor/src/pyinfinitensor/onnx.py b/pyinfinitensor/src/pyinfinitensor/onnx.py
index c8f5b1a5..c5c5638c 100644
--- a/pyinfinitensor/src/pyinfinitensor/onnx.py
+++ b/pyinfinitensor/src/pyinfinitensor/onnx.py
@@ -952,8 +952,8 @@ class OnnxStub:
                     size,
                 )
             elif node.op_type == "Dropout":
-                // HACK: Currently we only support model inference,
-                // so training_mode is set to false.
+                # HACK: Currently we only support model inference,
+                # so training_mode is set to false.
                 attributes = _parse_attribute(
                         node, {"ratio": 0.5, "training_mode": 0})
                 (ratio, training_mode) = (