Fix variable name issue