fix new introduced bug