fix mock client