Small fix for color picker

Still some bugs