Refactor: Change mock on real