Fix minor issue in the server