Small comment fix