Fix Python Tests.