We have been overriding the partners capital balance on Schedule L.  I have found no other way to do this.  Seems like it would be easy to code this computation by taking prior year capital plus book income per M-1/M-3, less distributions.