Fixing change from #111 in the source docs