Update keysForTree to match upstream.

Changes made upstream in
https://github.com/broccolijs/broccoli-kitchen-sink-helpers/pull/24.