Fix test broken in my last commit