Add missing include for function_traits.