minor fix

* update comments