fix set_color bug

fixes error
set_color: Expected an argument