Fix variable naming issue