下面的代码片段用于判断一个正整数是否为素数。请对以下代码进行修改，使其能正确实现相应功能。（   ）

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;
}