Small tweak to previous commit