fix #4

fix #4