Fix variable name from previous commit