update tests
fix python test