Fixed color assignment error