Fixed python unit test