Fixed an error in the code.