Fix the Server side socket

Not yet ran the junit, previous junit passed.