minor code fix

see the change