minor fix

minor fix after quick check