Add a small test in client

Add a small test in client