fix array shape in comment doc