Fixed a bug in the code.