Fix minor code error