Add warning parameter in tests