Small fix for 9778ab4