test(mock): mock fixed to contain used method