I'd say it's probably more the lack of sleep than any sort of punishment.

[ Even without hanging onto Cobb and making sure he didn't self destruct, dreamshare was a risky business. Arthur slept lightly, with a gun nearby, just in case. He'd made the mistake early of trusting his tracks were covered— never again. ]

Maybe, who knows? Might be hard to tell if they're a figment of the dream or not.