Partially fix problem with bad colours. Fix problem with unregistering cells in UI (why it worked earlier?)