The image above the example code does not correspond to code itself. This MR fixes it.
This is the image I am talking about: