Small tweak for #346