Fix error introduced by recent commit