Fix highlighted cells not refreshing