Fixed an error in the code