That seems a bit opaque and I try not to change what is reported on the K-1 that was provided by the S-Corp. It would work, but I'd hope there would be a better way to handle it.
