fix python tests?