Fix minor naming issue from last commit