Small fix for last patch