Fixed missing var in two places