minor fix based on feedback