Fix broken variable reference