Rename unit value to match AST