fix #1, add one line