fix source mistake

must fix #6