Revert back to mock instead of mockAs