Restore colors correctly