Fixed python tests