Fix code build error