Fix code problem