Update to use new variable name