minor fix and update