Fix some bugs in simple_zone.cu