punits: Clean up a bit.