Bug fix with return_cov