Fix UDP tests