Fix missed variable rename