Fix minor error in tests