Looks like this line should be updated to include the new variables.