fix #1

haha, just git contribute :p