Change mock setup