Fix a bug about arguments are not highlighted if it includes a break