fix #336

fix #336