Minor tweak to last commit