Fix test broken in #2388