diff --git a/pyloidal/cocos.py b/pyloidal/cocos.py index 1871b0c..e95cf55 100644 --- a/pyloidal/cocos.py +++ b/pyloidal/cocos.py @@ -148,10 +148,8 @@ def identify_cocos( if minor_radii is None: # Return both variants if not provided with minor radii return tuple( - itertools.chain.from_iterable( - sigma_to_cocos(sigma_bp, sigma_rpz, sigma_rtp, phi_by_2pi=x) + sigma_to_cocos(sigma_bp, sigma_rpz, sigma_rtp, psi_by_2pi=x) for x in (True, False) - ) ) index = np.argmin(np.abs(safety_factor))