Minor fix, use correct variable name