Fix error introduced in latest commit