Simple fix in client