Fix optional output type