fix current python unit test