Tiny fix to last commit