Fixes for is_aks_prime