minor fix - 2