Minor fix - Forgot to change variable names