refactor(mocks): simplify the  implementation