[FIX] PR comments fix