fix color on training type