Class diagram fix [ci skip]