minor fix and comment