Force Take() to get less than count if there are no enough elements

Enumerable.Take does do that automatically. Your code sample as given should work:

Take enumerates source and yields elements until count
elements have been yielded or source contains no more elements.