Fix code style bug