Fixed an error in last commit