We don't have tests for this behavior (my bad). Since you're modifying it, would you mind adding some to make everything still works here?