Fix regression introduced by #5