correct a bug for prime? for 2