They have specific instructions on this for Forms 5S and the Partnership returns on where to enter it.

For form 5S enter on line 10. Other additions: of Form 5S, Page 5-6.

I am filing extensions on these to avoid having to amend them if Wisconsin changes it's law. I heard they have introduced bills to correct this.