fixed index subset to not return at a sublevel when retrieving individual keys
fixed index subset to not return at a sublevel when retrieving individual keys

locuslab/dreaml,locuslab/dreaml,locuslab/dreaml