Forgot the edit one line

Fixed