Fix error in last commit