updated tests for client