Quick tweak to last commit