Fix variable name for class