minor fix

fixed bug