Updated client tests

fixed the client tests, theoretically it should all work fine