Fixed a code error