Update csi.go

fix infinite loop.