Fix color box disappearing on rotate