You are correct, you need to track the total of your capital balance plus non-recourse debt and reductions in that balance as you receive distributions.
See IRS instructions for K-1
