Correct.
You should have a separate account that reflects distributions to shareholders; an equity account.
This separate account essentially is tied to the retained earnings and would be summarized into this account on the balance sheet for the 1120-S tax return.
*A reminder that posts in a forum such as this do not constitute tax advice.
Also keep in mind the date of replies, as tax law changes.
