Adding minor fix