fix #0

add two lines in the end