Fix socket bugs