Addendum to #402

Fixes bug with un-annotated heads in TreebankRetriever