Fixed a bug

Fixed a bug which cause the variable name disappear. (line 758 and 837)