minor fix and ref