minor fix up