fix mock arguments