Fix code error