Fixed error as a result of variable rename.