Fixing the use of mocks vs real data