Fix Python unit tests
Fix another unit test