The study escapes me at the moment (it might be from a Dan Gilbert TED talk), but people who've just spent time on a difficult puzzle/task are also shown to have less willpower. It's not so much that we have a finite amount of willpower that we have a finite amount of cognitive resources that we can devote to various tasks, willpower being one of them.