Заменить материал вала с 40Х на один из данных (45Х, 50Х, 40XG). Произвести все необходимые расчеты на прочность и т.д.