Small fix to last commit