fix #335

fix #335