Fixed minor code error