Variable name change fix