Small fix after renaming