Fix oops introduced with last commit