Fix UDP case