Fix regression in custom colors function