下面的C++代码片段用于判断一个正整数是否为素数，请选择正确的修改方案使其能正确实现功能。

bool isPrime(int num) {
    if (num < 2) {
        return false;
    }
    for (int i = 2; i * i < num; ++i) {
        if (num % i == 0) {
            return false;
        }
    }
    return true;
}