fix #2

fix #2