fix COSMO unittest