Change two line, for this bug!