More test. Mock in place :D