Fixed a small error in the code.