Small tweak for #1