Fix little mistake

add some missing point in code