Add missing variable change