Fixed an error in previous commit