Fix line return bug