Small tweak to last change