fixed failed tests because mock setup was not updated