Update syntax in tests