Fix an oversight from the last commit