Small fix for my previous commit