我想把输入成绩的数量输错时，不用启动项目，可以再次输入，直到成功为止，咋整

import java.util.InputMismatchException;
import java.util.Scanner;

public class InputTest {
    public static void main(String[] args) {
        int num;
        Scanner scanner = new Scanner(System.in);
        while (true) {
            try {
                num = scanner.nextInt();
                break;
            } catch (InputMismatchException e) {
                scanner.next();
                System.out.println("输入不合法");
            }
        }
        System.out.println(num);
    }
}