bug fix

@vigsterkr 

Sorry for my mistake.
I forgot to record all the change in the previous PR.

i don't see any update... what update you mean?
i just see only one commit in this PR