Fix small bug in client