Fix Enumerable#chunk arg handling

Addresses the chunk part of #901