Fixed the description of input/output axes in fit method