@rubiojr fixed mock test failure.