Fix reference to old variable