Fix tests in test_performance