Fix a breaking change