Fix the broken tests.

Changing the variable names.

/cc @grosser 

### References

### Risks
 - None