fixing mock client