Small comment fix cuz of new code