Fixing missed variable name update