Fix incorrect variable reference