Fixed simple code error