fix new bug