Small fix for previous update