Fix collocations.doctest. See #399