Use proper assertion and fix to avoid colors overlapping