Fix test error added by 01afef8558d