Fix error introduced in last commit