Fix variable name bug