Made the socket test nicer