Small fix after last update