Small fix to my previous commit