minor fix + update