Fix optional inputs

@kitanata