Fix regression introduced in prior commit