Handle vertical alignment of 'for(...;' in token walker