Fix selected cell highlighting