Fix state colors