Added mock setup, so that all tests pass