You're probably thinking of ego depletion.