Small fix for previous commit